简介
通过向cerebro传入自定义Sizer,修改buy/sell的执行逻辑,控制下单量等。
在自定义的Sizer类中,可以访问策略对象self.strategy,以及策略的经纪行对象self.strategy.broker
(self.broker
),通过broker可以访问仓位(self.broker.getposition(data)
)和市值(self.broker.getvalue()
)等信息.
在Sizer类中的 def _getsizing(self, comminfo, cash, data, isbuy):
方法中,可通过其参数访问更多信息,包括佣金信息,现金,行情数据。
其中isbuy参数确定订单是买单还是卖单。该方法需要返回 int
类型表示下单量,如果返回 0。则不执行订单操作。
将自定义的Sizer注入Cerebro
有两种方法注入sizer到cerebro
1、所有策略都会使用该Sizer
cerebro.addsizer(LongOnly) # 全部策略使用自定义sizer
2、按策略id设置Sizer,给不同的策略设置不同的Sizer
idx = cerebro.addstrategy(SmaCross)
cerebro.addsizer_byidx(idx, maxRiskSizer)
示例
'''自定义sizer'''
import math
import backtrader as bt
from feed import feed
from logger import lg
class maxRiskSizer(bt.Sizer):
params = (
('risk', 0.03),
)
def __init__(self):
if self.p.risk > 1 or self.p.risk < 0:
raise ValueError(
'The risk parameter is a percentage which must be enter as a float: e.g. 0.5'<