1
0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-12-13 08:26:43 +00:00
This commit is contained in:
2020-05-02 01:23:11 +08:00
parent e77537b553
commit 2e803f0ec9
2 changed files with 15 additions and 14 deletions

View File

@@ -117,10 +117,10 @@ class Data:
test_x = [feature_data[self.start_num_in_test + i * self.config.time_step: self.start_num_in_test + (
i + 1) * self.config.time_step]
for i in range(time_step_size)]
print("test_x is", test_x)
if return_label_data:
label_data = self.norm_data[self.train_num + self.start_num_in_test:, self.config.label_in_feature_columns]
return np.array(test_x), label_data
print(test_x)
return np.array(test_x)
# add yqy