liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-12 15:26:54 +08:00
Code Activity

supporting seperated buy and sell price

Browse Source
This commit is contained in:
Young
2021-07-06 06:28:14 +00:00
parent cb72857710
commit bdac9f4dda
3 changed files with 75 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
qlib/contrib/strategy/model_strategy.py
View File

@@ -6,7 +6,7 @@ import pandas as pd
from ...utils.resam import resam_ts_data
from ...strategy.base import ModelStrategy
from ...backtest.order import Order, BaseTradeDecision, TradeDecisionWO
from ...backtest.order import Order, BaseTradeDecision, OrderDir, TradeDecisionWO
from .order_generator import OrderGenWInteract
@@ -236,7 +236,7 @@ class TopkDropoutStrategy(ModelStrategy):
continue
# buy order
buy_price = self.trade_exchange.get_deal_price(
stock_id=code, start_time=trade_start_time, end_time=trade_end_time
stock_id=code, start_time=trade_start_time, end_time=trade_end_time, direction=OrderDir.BUY
)
buy_amount = value / buy_price
factor = self.trade_exchange.get_factor(stock_id=code, start_time=trade_start_time, end_time=trade_end_time)