Commit
·
9c0b92a
1
Parent(s):
1d22072
test5
Browse files
app.py
CHANGED
@@ -19,14 +19,14 @@ def prepare_data_chronos(data):
|
|
19 |
formatted_df = pd.DataFrame({
|
20 |
'item_id': ['stock'] * len(df),
|
21 |
'timestamp': pd.to_datetime(df['Date']),
|
22 |
-
'
|
23 |
})
|
24 |
|
25 |
# 按照時間戳排序
|
26 |
formatted_df = formatted_df.sort_values('timestamp')
|
27 |
|
28 |
try:
|
29 |
-
#
|
30 |
ts_df = TimeSeriesDataFrame.from_data_frame(
|
31 |
formatted_df,
|
32 |
id_column='item_id',
|
@@ -116,10 +116,11 @@ def get_top_10_potential_stocks(period, selected_indices):
|
|
116 |
|
117 |
ts_data = prepare_data_chronos(data)
|
118 |
|
119 |
-
#
|
120 |
predictor = TimeSeriesPredictor(
|
121 |
prediction_length=prediction_length,
|
122 |
-
freq="
|
|
|
123 |
)
|
124 |
|
125 |
predictor.fit(
|
|
|
19 |
formatted_df = pd.DataFrame({
|
20 |
'item_id': ['stock'] * len(df),
|
21 |
'timestamp': pd.to_datetime(df['Date']),
|
22 |
+
'target': df['Close'].astype('float32').values.ravel() # 改回使用 'target' 而不是 'value'
|
23 |
})
|
24 |
|
25 |
# 按照時間戳排序
|
26 |
formatted_df = formatted_df.sort_values('timestamp')
|
27 |
|
28 |
try:
|
29 |
+
# 創建 TimeSeriesDataFrame
|
30 |
ts_df = TimeSeriesDataFrame.from_data_frame(
|
31 |
formatted_df,
|
32 |
id_column='item_id',
|
|
|
116 |
|
117 |
ts_data = prepare_data_chronos(data)
|
118 |
|
119 |
+
# 創建預測器,使用 'target' 作為目標列名
|
120 |
predictor = TimeSeriesPredictor(
|
121 |
prediction_length=prediction_length,
|
122 |
+
freq="D", # 使用 'D' 而不是 '1D'
|
123 |
+
target="target" # 確保與數據框中的列名一致
|
124 |
)
|
125 |
|
126 |
predictor.fit(
|