V2 与 V3 之间的策略迁移 (Strategy Migration)
为了支持新的市场和交易类型(即做空交易 / 杠杆交易),接口中必须进行一些更改。 如果您打算使用现货市场以外的市场,请将您的策略迁移到新格式。
我们投入了大量精力来保持与现有策略的兼容性,因此如果您只想继续在 现货市场 中使用 Freqtrade,目前应该无需进行任何更改。
您可以将快速摘要作为清单。请参阅下面的详细章节了解完整的迁移详情。
快速摘要 / 迁移清单
注意:forcesell、forcebuy、emergencysell 分别更改为 force_exit、force_enter、emergency_exit。
- 策略方法:
- 数据帧 (Dataframe) 列:
- trade 对象现在具有以下新属性:
is_shortentry_sideexit_sidetrade_direction- 已重命名:
sell_reason->exit_reason
- 将
trade.nr_of_successful_buys重命名为trade.nr_of_successful_entries(主要与adjust_trade_position()相关) - 引入了新的
leverage回调。 - 信息性交易对 (Informative pairs) 现在可以在元组中传递第 3 个元素,定义 K 线类型。
@informative装饰器现在接受可选的candle_type参数。- 辅助方法
stoploss_from_open和stoploss_from_absolute现在接受is_short作为额外参数。 INTERFACE_VERSION应设置为 3。- 策略/配置设置。
order_time_in_forcebuy -> entry, sell -> exit。order_typesbuy -> entry, sell -> exit。unfilledtimeoutbuy -> entry, sell -> exit。ignore_buying_expired_candle_after-> 移动到根级别,不再位于 "ask_strategy/exit_pricing" 下。
- 术语变更
- 卖出原因 (Sell reasons) 更改为反映 "exit"(离场)而不是 sells。如果您的策略使用了
exit_reason检查,请务必小心并更新策略。sell_signal->exit_signalcustom_sell->custom_exitforce_sell->force_exitemergency_sell->emergency_exit
- 订单定价 (Order pricing)
bid_strategy->entry_pricingask_strategy->exit_pricingask_last_balance->price_last_balancebid_last_balance->price_last_balance
- Webhook 术语从 "sell" 更改为 "exit",从 "buy" 更改为 "entry"
webhookbuy->entrywebhookbuyfill->entry_fillwebhookbuycancel->entry_cancelwebhooksell->exitwebhooksellfill->exit_fillwebhooksellcancel->exit_cancel
- Telegram 通知设置
buy->entrybuy_fill->entry_fillbuy_cancel->entry_cancelsell->exitsell_fill->exit_fillsell_cancel->exit_cancel
- 策略/配置设置:
use_sell_signal->use_exit_signalsell_profit_only->exit_profit_onlysell_profit_offset->exit_profit_offsetignore_roi_if_buy_signal->ignore_roi_if_entry_signalforcebuy_enable->force_entry_enable
- 卖出原因 (Sell reasons) 更改为反映 "exit"(离场)而不是 sells。如果您的策略使用了
详细说明
populate_buy_trend
在 populate_buy_trend() 中 —— 您需要将赋值的列从 'buy' 更改为 'enter_long',并将方法名从 populate_buy_trend 更改为 populate_entry_trend。
def populate_buy_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
dataframe.loc[
(
(qtpylib.crossed_above(dataframe['rsi'], 30)) & # 信号:RSI 向上穿越 30
(dataframe['tema'] <= dataframe['bb_middleband']) & # 守护条件
(dataframe['tema'] > dataframe['tema'].shift(1)) & # 守护条件
(dataframe['volume'] > 0) # 确保交易量不为 0
),
['buy', 'buy_tag']] = (1, 'rsi_cross')
return dataframe迁移后:
def populate_entry_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
dataframe.loc[
(
(qtpylib.crossed_above(dataframe['rsi'], 30)) & # 信号:RSI 向上穿越 30
(dataframe['tema'] <= dataframe['bb_middleband']) & # 守护条件
(dataframe['tema'] > dataframe['tema'].shift(1)) & # 守护条件
(dataframe['volume'] > 0) # 确保交易量不为 0
),
['enter_long', 'enter_tag']] = (1, 'rsi_cross')
return dataframe请参阅 策略文档 了解如何进场和离场做空交易。
populate_sell_trend
与 populate_buy_trend 类似,populate_sell_trend() 将重命名为 populate_exit_trend()。 我们还将列名从 'sell' 更改为 'exit_long'。
def populate_sell_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
dataframe.loc[
(
(qtpylib.crossed_above(dataframe['rsi'], 70)) & # 信号:RSI 向上穿越 70
(dataframe['tema'] > dataframe['bb_middleband']) & # 守护条件
(dataframe['tema'] < dataframe['tema'].shift(1)) & # 守护条件
(dataframe['volume'] > 0) # 确保交易量不为 0
),
['sell', 'exit_tag']] = (1, 'some_exit_tag')
return dataframe迁移后:
def populate_exit_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
dataframe.loc[
(
(qtpylib.crossed_above(dataframe['rsi'], 70)) & # 信号:RSI 向上穿越 70
(dataframe['tema'] > dataframe['bb_middleband']) & # 守护条件
(dataframe['tema'] < dataframe['tema'].shift(1)) & # 守护条件
(dataframe['volume'] > 0) # 确保交易量不为 0
),
['exit_long', 'exit_tag']] = (1, 'some_exit_tag')
return dataframe请参阅 策略文档 了解如何进场和离场做空交易。
custom_sell
custom_sell 已重命名为 custom_exit。 它现在在每次迭代时都会被调用,不依赖于当前利润和 exit_profit_only 设置。
class AwesomeStrategy(IStrategy):
def custom_sell(self, pair: str, trade: 'Trade', current_time: 'datetime', current_rate: float,
current_profit: float, **kwargs):
dataframe, _ = self.dp.get_analyzed_dataframe(pair, self.timeframe)
last_candle = dataframe.iloc[-1].squeeze()
# ...class AwesomeStrategy(IStrategy):
def custom_exit(self, pair: str, trade: 'Trade', current_time: 'datetime', current_rate: float,
current_profit: float, **kwargs):
dataframe, _ = self.dp.get_analyzed_dataframe(pair, self.timeframe)
last_candle = dataframe.iloc[-1].squeeze()
# ...custom_entry_timeout
check_buy_timeout() 已重命名为 check_entry_timeout(),check_sell_timeout() 已重命名为 check_exit_timeout()。
class AwesomeStrategy(IStrategy):
def check_buy_timeout(self, pair: str, trade: 'Trade', order: dict,
current_time: datetime, **kwargs) -> bool:
return False
def check_sell_timeout(self, pair: str, trade: 'Trade', order: dict,
current_time: datetime, **kwargs) -> bool:
return Falseclass AwesomeStrategy(IStrategy):
def check_entry_timeout(self, pair: str, trade: 'Trade', order: 'Order',
current_time: datetime, **kwargs) -> bool:
return False
def check_exit_timeout(self, pair: str, trade: 'Trade', order: 'Order',
current_time: datetime, **kwargs) -> bool:
return Falsecustom_stake_amount
新增字符串参数 side —— 值为 "long" 或 "short"。
class AwesomeStrategy(IStrategy):
def custom_stake_amount(self, pair: str, current_time: datetime, current_rate: float,
proposed_stake: float, min_stake: Optional[float], max_stake: float,
entry_tag: Optional[str], **kwargs) -> float:
# ...
return proposed_stakeclass AwesomeStrategy(IStrategy):
def custom_stake_amount(self, pair: str, current_time: datetime, current_rate: float,
proposed_stake: float, min_stake: float | None, max_stake: float,
entry_tag: str | None, side: str, **kwargs) -> float:
# ...
return proposed_stakeconfirm_trade_entry
新增字符串参数 side —— 值为 "long" 或 "short"。
class AwesomeStrategy(IStrategy):
def confirm_trade_entry(self, pair: str, order_type: str, amount: float, rate: float,
time_in_force: str, current_time: datetime, entry_tag: Optional[str],
**kwargs) -> bool:
return True迁移后:
class AwesomeStrategy(IStrategy):
def confirm_trade_entry(self, pair: str, order_type: str, amount: float, rate: float,
time_in_force: str, current_time: datetime, entry_tag: str | None,
side: str, **kwargs) -> bool:
return Trueconfirm_trade_exit
将参数 sell_reason 更改为 exit_reason。 为了保持兼容性,sell_reason 在短时间内仍会提供。
class AwesomeStrategy(IStrategy):
def confirm_trade_exit(self, pair: str, trade: Trade, order_type: str, amount: float,
rate: float, time_in_force: str, sell_reason: str,
current_time: datetime, **kwargs) -> bool:
return True迁移后:
class AwesomeStrategy(IStrategy):
def confirm_trade_exit(self, pair: str, trade: Trade, order_type: str, amount: float,
rate: float, time_in_force: str, exit_reason: str,
current_time: datetime, **kwargs) -> bool:
return Truecustom_entry_price
新增字符串参数 side —— 值为 "long" 或 "short"。
class AwesomeStrategy(IStrategy):
def custom_entry_price(self, pair: str, current_time: datetime, proposed_rate: float,
entry_tag: Optional[str], **kwargs) -> float:
return proposed_rate迁移后:
class AwesomeStrategy(IStrategy):
def custom_entry_price(self, pair: str, trade: Trade | None, current_time: datetime, proposed_rate: float,
entry_tag: str | None, side: str, **kwargs) -> float:
return proposed_rate调整交易仓位 (adjust-trade-position) 的更改
虽然 adjust-trade-position 本身没有改变,但您不应再使用 trade.nr_of_successful_buys —— 而应使用 trade.nr_of_successful_entries(重命名后包含了做空进场)。
辅助方法
为 stoploss_from_open 和 stoploss_from_absolute 添加了参数 "is_short"。 应传入 trade.is_short 的值。
def custom_stoploss(self, pair: str, trade: 'Trade', current_time: datetime,
current_rate: float, current_profit: float, **kwargs) -> float:
# 一旦利润超过 10%,将止损设置在开仓价格上方 7% 处
if current_profit > 0.10:
return stoploss_from_open(0.07, current_profit)
return stoploss_from_absolute(current_rate - (candle['atr'] * 2), current_rate)
return 1迁移后:
def custom_stoploss(self, pair: str, trade: 'Trade', current_time: datetime,
current_rate: float, current_profit: float, after_fill: bool,
**kwargs) -> float | None:
# 一旦利润超过 10%,将止损设置在开仓价格上方 7% 处
if current_profit > 0.10:
return stoploss_from_open(0.07, current_profit, is_short=trade.is_short)
return stoploss_from_absolute(current_rate - (candle['atr'] * 2), current_rate, is_short=trade.is_short, leverage=trade.leverage)策略/配置设置
order_time_in_force
order_time_in_force 属性从 "buy" 更改为 "entry",从 "sell" 更改为 "exit"。
order_time_in_force: dict = {
"buy": "gtc",
"sell": "gtc",
}迁移后:
order_time_in_force: dict = {
"entry": "GTC",
"exit": "GTC",
}order_types
order_types 将所有词汇从 buy 更改为 entry —— 从 sell 更改为 exit。 并且两个单词通过 _ 连接。
order_types = {
"buy": "limit",
"sell": "limit",
"emergencysell": "market",
"forcesell": "market",
"forcebuy": "market",
"stoploss": "market",
"stoploss_on_exchange": false,
"stoploss_on_exchange_interval": 60
}迁移后:
order_types = {
"entry": "limit",
"exit": "limit",
"emergency_exit": "market",
"force_exit": "market",
"force_entry": "market",
"stoploss": "market",
"stoploss_on_exchange": false,
"stoploss_on_exchange_interval": 60
}策略级别设置
use_sell_signal->use_exit_signalsell_profit_only->exit_profit_onlysell_profit_offset->exit_profit_offsetignore_roi_if_buy_signal->ignore_roi_if_entry_signal
# 这些值可以在配置中覆盖。
use_sell_signal = True
sell_profit_only = True
sell_profit_offset: 0.01
ignore_roi_if_buy_signal = False迁移后:
# 这些值可以在配置中覆盖。
use_exit_signal = True
exit_profit_only = True
exit_profit_offset: 0.01
ignore_roi_if_entry_signal = Falseunfilledtimeout
unfilledtimeout 将所有词汇从 buy 更改为 entry —— 从 sell 更改为 exit。
unfilledtimeout = {
"buy": 10,
"sell": 10,
"exit_timeout_count": 0,
"unit": "minutes"
}迁移后:
unfilledtimeout = {
"entry": 10,
"exit": 10,
"exit_timeout_count": 0,
"unit": "minutes"
}订单定价 (order pricing)
订单定价有两个维度的变化。bid_strategy 重命名为 entry_pricing,ask_strategy 重命名为 exit_pricing。 属性 ask_last_balance -> price_last_balance 以及 bid_last_balance -> price_last_balance 也被重命名。 此外,price-side 现在可以定义为 ask、bid、same 或 other。 请参阅 定价文档 了解更多信息。
{
"bid_strategy": {
"price_side": "bid",
"use_order_book": true,
"order_book_top": 1,
"ask_last_balance": 0.0,
"check_depth_of_market": {
"enabled": false,
"bids_to_ask_delta": 1
}
},
"ask_strategy":{
"price_side": "ask",
"use_order_book": true,
"order_book_top": 1,
"bid_last_balance": 0.0
"ignore_buying_expired_candle_after": 120
}
}迁移后:
{
"entry_pricing": {
"price_side": "same",
"use_order_book": true,
"order_book_top": 1,
"price_last_balance": 0.0,
"check_depth_of_market": {
"enabled": false,
"bids_to_ask_delta": 1
}
},
"exit_pricing":{
"price_side": "same",
"use_order_book": true,
"order_book_top": 1,
"price_last_balance": 0.0
},
"ignore_buying_expired_candle_after": 120
}FreqAI 策略
populate_any_indicators() 方法已被拆分为 feature_engineering_expand_all()、feature_engineering_expand_basic()、feature_engineering_standard() 和 set_freqai_targets()。
对于每个新函数,交易对(必要时还包括时间框架)将自动添加到列名中。 因此,使用新逻辑定义特征变得更加简单。
有关每个方法的详细解释,请访问相应的 FreqAI 文档页面。
def populate_any_indicators(
self, pair, df, tf, informative=None, set_generalized_indicators=False
):
if informative is None:
informative = self.dp.get_pair_dataframe(pair, tf)
# 第一个循环自动为时间段复制指标
for t in self.freqai_info["feature_parameters"]["indicator_periods_candles"]:
t = int(t)
informative[f"%-{pair}rsi-period_{t}"] = ta.RSI(informative, timeperiod=t)
informative[f"%-{pair}mfi-period_{t}"] = ta.MFI(informative, timeperiod=t)
informative[f"%-{pair}adx-period_{t}"] = ta.ADX(informative, timeperiod=t)
informative[f"%-{pair}sma-period_{t}"] = ta.SMA(informative, timeperiod=t)
informative[f"%-{pair}ema-period_{t}"] = ta.EMA(informative, timeperiod=t)
bollinger = qtpylib.bollinger_bands(
qtpylib.typical_price(informative), window=t, stds=2.2
)
informative[f"{pair}bb_lowerband-period_{t}"] = bollinger["lower"]
informative[f"{pair}bb_middleband-period_{t}"] = bollinger["mid"]
informative[f"{pair}bb_upperband-period_{t}"] = bollinger["upper"]
informative[f"%-{pair}bb_width-period_{t}"] = (
informative[f"{pair}bb_upperband-period_{t}"]
- informative[f"{pair}bb_lowerband-period_{t}"]
) / informative[f"{pair}bb_middleband-period_{t}"]
informative[f"%-{pair}close-bb_lower-period_{t}"] = (
informative["close"] / informative[f"{pair}bb_lowerband-period_{t}"]
)
informative[f"%-{pair}roc-period_{t}"] = ta.ROC(informative, timeperiod=t)
informative[f"%-{pair}relative_volume-period_{t}"] = (
informative["volume"] / informative["volume"].rolling(t).mean()
) # (1)
informative[f"%-{pair}pct-change"] = informative["close"].pct_change()
informative[f"%-{pair}raw_volume"] = informative["volume"]
informative[f"%-{pair}raw_price"] = informative["close"]
# (2)
indicators = [col for col in informative if col.startswith("%")]
# 此循环对所有指标进行复制和移动,以为数据添加近期感
for n in range(self.freqai_info["feature_parameters"]["include_shifted_candles"] + 1):
if n == 0:
continue
informative_shift = informative[indicators].shift(n)
informative_shift = informative_shift.add_suffix("_shift-" + str(n))
informative = pd.concat((informative, informative_shift), axis=1)
df = merge_informative_pair(df, informative, self.config["timeframe"], tf, ffill=True)
skip_columns = [
(s + "_" + tf) for s in ["date", "open", "high", "low", "close", "volume"]
]
df = df.drop(columns=skip_columns)
# 在此处添加通用指标(因为在实盘中,它会在训练期间调用此函数
# 来填充指标)。注意我们如何确保不重复添加它们
if set_generalized_indicators:
df["%-day_of_week"] = (df["date"].dt.dayofweek + 1) / 7
df["%-hour_of_day"] = (df["date"].dt.hour + 1) / 25
# (3)
# 用户在此处添加目标,通过在前缀加上 &- (参见下方惯例)
df["&-s_close"] = (
df["close"]
.shift(-self.freqai_info["feature_parameters"]["label_period_candles"])
.rolling(self.freqai_info["feature_parameters"]["label_period_candles"])
.mean()
/ df["close"]
- 1
) # (4)
return df- 特征 —— 移动到
feature_engineering_expand_all。 - 基础特征,不随
indicator_periods_candles展开 —— 移动到feature_engineering_expand_basic()。 - 不应展开的标准特征 —— 移动到
feature_engineering_standard()。 - 目标 —— 将此部分移动到
set_freqai_targets()。
FreqAI - 特征工程全部展开 (feature engineering expand all)
特征现在会自动展开。因此,需要移除展开循环以及 {pair} / {timeframe} 部分。
def feature_engineering_expand_all(self, dataframe, period, **kwargs) -> DataFrame::
"""
*仅对启用 FreqAI 的策略有效*
此函数将根据配置中定义的 `indicator_periods_candles`、`include_timeframes`、
`include_shifted_candles` 和 `include_corr_pairs` 自动展开定义的特征。
换句话说,在此函数中定义的一个特征将自动展开,总共添加
`indicator_periods_candles` * `include_timeframes` * `include_shifted_candles` *
`include_corr_pairs` 数量的特征到模型中。
所有特征必须前缀 `%` 才能被 FreqAI 内部识别。
有关这些配置定义参数如何加速特征工程的更多详情,请参见文档:
https://www.freqtrade.io/en/latest/freqai-parameter-table/#feature-parameters
https://www.freqtrade.io/en/latest/freqai-feature-engineering/#defining-the-features
:param df: 接收特征的策略数据帧
:param period: 指标的周期 - 用法示例:
dataframe["%-ema-period"] = ta.EMA(dataframe, timeperiod=period)
"""
dataframe["%-rsi-period"] = ta.RSI(dataframe, timeperiod=period)
dataframe["%-mfi-period"] = ta.MFI(dataframe, timeperiod=period)
dataframe["%-adx-period"] = ta.ADX(dataframe, timeperiod=period)
dataframe["%-sma-period"] = ta.SMA(dataframe, timeperiod=period)
dataframe["%-ema-period"] = ta.EMA(dataframe, timeperiod=period)
bollinger = qtpylib.bollinger_bands(
qtpylib.typical_price(dataframe), window=period, stds=2.2
)
dataframe["bb_lowerband-period"] = bollinger["lower"]
dataframe["bb_middleband-period"] = bollinger["mid"]
dataframe["bb_upperband-period"] = bollinger["upper"]
dataframe["%-bb_width-period"] = (
dataframe["bb_upperband-period"]
- dataframe["bb_lowerband-period"]
) / dataframe["bb_middleband-period"]
dataframe["%-close-bb_lower-period"] = (
dataframe["close"] / dataframe["bb_lowerband-period"]
)
dataframe["%-roc-period"] = ta.ROC(dataframe, timeperiod=period)
dataframe["%-relative_volume-period"] = (
dataframe["volume"] / dataframe["volume"].rolling(period).mean()
)
return dataframeFreqAI - 特征工程基础 (feature engineering basic)
基础特征。确保从特征中移除 {pair} 部分。
def feature_engineering_expand_basic(self, dataframe: DataFrame, **kwargs) -> DataFrame::
"""
*仅对启用 FreqAI 的策略有效*
此函数将根据配置中定义的 `include_timeframes`、`include_shifted_candles`
和 `include_corr_pairs` 自动展开定义的特征。
换句话说,在此函数中定义的一个特征将自动展开,总共添加
`include_timeframes` * `include_shifted_candles` * `include_corr_pairs`
数量的特征到模型中。
此处定义的特征 *不会* 随用户定义的 `indicator_periods_candles` 自动复制。
所有特征必须前缀 `%` 才能被 FreqAI 内部识别。
有关这些配置定义参数如何加速特征工程的更多详情,请参见文档:
https://www.freqtrade.io/en/latest/freqai-parameter-table/#feature-parameters
https://www.freqtrade.io/en/latest/freqai-feature-engineering/#defining-the-features
:param df: 接收特征的策略数据帧
dataframe["%-pct-change"] = dataframe["close"].pct_change()
dataframe["%-ema-200"] = ta.EMA(dataframe, timeperiod=200)
"""
dataframe["%-pct-change"] = dataframe["close"].pct_change()
dataframe["%-raw_volume"] = dataframe["volume"]
dataframe["%-raw_price"] = dataframe["close"]
return dataframeFreqAI - 特征工程标准 (feature engineering standard)
def feature_engineering_standard(self, dataframe: DataFrame, **kwargs) -> DataFrame:
"""
*仅对启用 FreqAI 的策略有效*
此可选函数将随基础时间框架的数据帧被调用一次。
这是最后一个被调用的函数,这意味着进入此函数的数据帧将包含由所有其他
freqai_feature_engineering_* 函数创建的所有特征和列。
此函数是进行自定义异域特征提取(例如 tsfresh)的好地方。
此函数同样适合任何不应被自动展开的特征(例如星期几)。
所有特征必须前缀 `%` 才能被 FreqAI 内部识别。
有关可用特征工程的更多详情:
https://www.freqtrade.io/en/latest/freqai-feature-engineering
:param df: 接收特征的策略数据帧
用法示例:dataframe["%-day_of_week"] = (dataframe["date"].dt.dayofweek + 1) / 7
"""
dataframe["%-day_of_week"] = dataframe["date"].dt.dayofweek
dataframe["%-hour_of_day"] = dataframe["date"].dt.hour
return dataframeFreqAI - 设置目标 (set Targets)
目标现在有了自己专用的方法。
def set_freqai_targets(self, dataframe: DataFrame, **kwargs) -> DataFrame:
"""
*仅对启用 FreqAI 的策略有效*
设置模型目标的必要函数。
所有目标必须前缀 `&` 才能被 FreqAI 内部识别。
有关可用特征工程的更多详情:
https://www.freqtrade.io/en/latest/freqai-feature-engineering
:param df: 接收目标的策略数据帧
用法示例:dataframe["&-target"] = dataframe["close"].shift(-1) / dataframe["close"]
"""
dataframe["&-s_close"] = (
dataframe["close"]
.shift(-self.freqai_info["feature_parameters"]["label_period_candles"])
.rolling(self.freqai_info["feature_parameters"]["label_period_candles"])
.mean()
/ dataframe["close"]
- 1
)
return dataframeFreqAI - 新数据流水线 (New data Pipeline)
如果您创建了具有自定义 train()/predict() 函数的自定义 IFreqaiModel,并且 您仍然依赖 data_cleaning_train/predict(),那么您需要迁移到新的流水线。如果您的模型 不 依赖 data_cleaning_train/predict(),那么您无需担心此迁移。这意味着此迁移指南仅对极少数高级用户有用。如果您误入此指南,欢迎在 Freqtrade Discord 服务器中深入查询您的问题。
转换涉及首先移除 data_cleaning_train/predict(),并在您的 IFreqaiModel 类中替换为 define_data_pipeline() 和 define_label_pipeline() 函数:
class MyCoolFreqaiModel(BaseRegressionModel):
"""
您在 Freqtrade 2023.6 版本之前编写的一些酷炫自定义 IFreqaiModel
"""
def train(
self, unfiltered_df: DataFrame, pair: str, dk: FreqaiDataKitchen, **kwargs
) -> Any:
# ... 您自定义的内容
# 移除这些行
# data_dictionary = dk.make_train_test_datasets(features_filtered, labels_filtered)
# self.data_cleaning_train(dk)
# data_dictionary = dk.normalize_data(data_dictionary)
# (1)
# 添加这些行。现在我们自己控制 pipeline 的 fit/transform
dd = dk.make_train_test_datasets(features_filtered, labels_filtered)
dk.feature_pipeline = self.define_data_pipeline(threads=dk.thread_count)
dk.label_pipeline = self.define_label_pipeline(threads=dk.thread_count)
(dd["train_features"],
dd["train_labels"],
dd["train_weights"]) = dk.feature_pipeline.fit_transform(dd["train_features"],
dd["train_labels"],
dd["train_weights"])
(dd["test_features"],
dd["test_labels"],
dd["test_weights"]) = dk.feature_pipeline.transform(dd["test_features"],
dd["test_labels"],
dd["test_weights"])
dd["train_labels"], _, _ = dk.label_pipeline.fit_transform(dd["train_labels"])
dd["test_labels"], _, _ = dk.label_pipeline.transform(dd["test_labels"])
# ... 您自定义的代码
return model
def predict(
self, unfiltered_df: DataFrame, dk: FreqaiDataKitchen, **kwargs
) -> tuple[DataFrame, npt.NDArray[np.int_]]:
# ... 您自定义的内容
# 移除这些行:
# self.data_cleaning_predict(dk)
# (2)
# 添加这些行:
dk.data_dictionary["prediction_features"], outliers, _ = dk.feature_pipeline.transform(
dk.data_dictionary["prediction_features"], outlier_check=True)
# 移除这一行
# pred_df = dk.denormalize_labels_from_metadata(pred_df)
# (3)
# 替换为这些行
pred_df, _, _ = dk.label_pipeline.inverse_transform(pred_df)
if self.freqai_info.get("DI_threshold", 0) > 0:
dk.DI_values = dk.feature_pipeline["di"].di_values
else:
dk.DI_values = np.zeros(outliers.shape[0])
dk.do_predict = outliers
# ... 您自定义的代码
return (pred_df, dk.do_predict)- 数据归一化和清洗现在已与新的流水线定义统一。这是在新的
define_data_pipeline()和define_label_pipeline()函数中创建的。data_cleaning_train()和data_cleaning_predict()函数已不再使用。如果愿意,您可以覆盖define_data_pipeline()以创建自己的自定义流水线。 - 数据归一化和清洗现在已与新的流水线定义统一。这是在新的
define_data_pipeline()和define_label_pipeline()函数中创建的。data_cleaning_train()和data_cleaning_predict()函数已不再使用。如果愿意,您可以覆盖define_data_pipeline()以创建自己的自定义流水线。 - 数据反归一化通过新的流水线完成。替换为下方的行。
