Skip to content

R
robo-dividend
Commit c0a487e0 authored by wenwen.tang's avatar wenwen.tang 😕
Browse files
Options

add print log

parent 6d823536
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
ai/EstimateMarketTrendV20.py
View file @ c0a487e0
...@@ -102,6 +102,7 @@ if __name__ == '__main__': ...@@ -102,6 +102,7 @@ if __name__ == '__main__':
builder = TrainingDataBuilder(index, eco, fund, indexDict, toForecast, win1W, win1M, win1Q, numForecastDays, builder = TrainingDataBuilder(index, eco, fund, indexDict, toForecast, win1W, win1M, win1Q, numForecastDays,
theThreshold) theThreshold)
for pid in PREDICT_LIST: for pid in PREDICT_LIST:
print(f'{indexDict[pid]} start '.center(50, '='))
t_data = indexData if pid in index else fundData t_data = indexData if pid in index else fundData
X_train, X_test, y_train, y_test, scaledX_forecast, forecastDay = \ X_train, X_test, y_train, y_test, scaledX_forecast, forecastDay = \
builder.build_train_test(pid, t_data, vixData, indexOtherData, cpiData, FDTRData, NAPMPMIData) builder.build_train_test(pid, t_data, vixData, indexOtherData, cpiData, FDTRData, NAPMPMIData)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment