fix: align pooled symbol features at training
This commit is contained in:
@@ -361,6 +361,7 @@ def _train_pooled_symbols(
|
||||
prepared_by_symbol: dict[str, PreparedData] = {}
|
||||
for symbol in symbols:
|
||||
prepared = _prepare_data(
|
||||
symbol=symbol,
|
||||
candles=market_candles[symbol],
|
||||
feature_names=feature_names,
|
||||
lookback=lookback,
|
||||
@@ -654,6 +655,7 @@ def _train_symbol(
|
||||
for lookback in lookbacks:
|
||||
_progress(f"{symbol}: preparing lookback={lookback}")
|
||||
prepared = _prepare_data(
|
||||
symbol=symbol,
|
||||
candles=candles,
|
||||
feature_names=feature_names,
|
||||
lookback=lookback,
|
||||
@@ -777,6 +779,7 @@ def _train_symbol(
|
||||
|
||||
def _prepare_data(
|
||||
*,
|
||||
symbol: str,
|
||||
candles: list[Candle],
|
||||
feature_names: list[str],
|
||||
lookback: int,
|
||||
@@ -796,6 +799,7 @@ def _prepare_data(
|
||||
feature_rows = _feature_matrix(
|
||||
candles,
|
||||
feature_names,
|
||||
symbol=symbol,
|
||||
market_candles=market_candles,
|
||||
trend_candles=trend_candles,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user