Commit67b84d8

committed

fix loading data on some tickers on stock prediction tutorial

1 parent1151b59 commit67b84d8Copy full SHA for 67b84d8

File tree

-0

lines changed

-0

lines changed

Lines changed: 2 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -215,6 +215,8 @@`
`215`	`215`	`" dates = result[\"X_test\"][:, -1, -1]\n",`
`216`	`216`	`" # retrieve test features from the original dataframe\n",`
`217`	`217`	`" result[\"test_df\"] = result[\"df\"].loc[dates]\n",`
	`218`	`+" # remove duplicated dates in the testing dataframe\n",`
	`219`	`+" result[\"test_df\"] = result[\"test_df\"][~result[\"test_df\"].index.duplicated(keep='first')]\n",`
`218`	`220`	`" # remove dates from the training/testing sets & convert to float32\n",`
`219`	`221`	`" result[\"X_train\"] = result[\"X_train\"][:, :, :len(feature_columns)].astype(np.float32)\n",`
`220`	`222`	`" result[\"X_test\"] = result[\"X_test\"][:, :, :len(feature_columns)].astype(np.float32)\n",`

Lines changed: 2 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -129,6 +129,8 @@ def load_data(ticker, n_steps=50, scale=True, shuffle=True, lookup_step=1, split`
`129`	`129`	`dates=result["X_test"][:,-1,-1]`
`130`	`130`	`# retrieve test features from the original dataframe`
`131`	`131`	`result["test_df"]=result["df"].loc[dates]`
	`132`	`+# remove duplicated dates in the testing dataframe`
	`133`	`+result["test_df"]=result["test_df"][~result["test_df"].index.duplicated(keep='first')]`
`132`	`134`	`# remove dates from the training/testing sets & convert to float32`
`133`	`135`	`result["X_train"]=result["X_train"][:, :, :len(feature_columns)].astype(np.float32)`
`134`	`136`	`result["X_test"]=result["X_test"][:, :, :len(feature_columns)].astype(np.float32)`

Comments

(0)