YAO/YAO-optimizer
1
0
Fork 0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-06-07 15:11:56 +00:00
Code Issues Releases Wiki Activity

add files

Browse Source
This commit is contained in:
Newnius 2020-04-30 01:10:50 +08:00
parent 0560bca8f6
commit 8a6fecbb3b
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
serve.py
View File

@ -51,6 +51,7 @@ class Config:
continue_flag = "continue_" continue_flag = "continue_"
train_data_path = "./data.csv" train_data_path = "./data.csv"
test_data_path = "./test.csv"
model_save_path = "./checkpoint/" model_save_path = "./checkpoint/"
figure_save_path = "./figure/" figure_save_path = "./figure/"
do_figure_save = False do_figure_save = False
@ -112,7 +113,7 @@ class Data:
def get_test_data(self, return_label_data=False): def get_test_data(self, return_label_data=False):
init_data = pd.read_csv( init_data = pd.read_csv(
self.config.train_data_path, self.config.test_data_path,
usecols=self.config.feature_and_label_columns usecols=self.config.feature_and_label_columns
) )
data, data_column_name = init_data.values, init_data.columns.tolist() data, data_column_name = init_data.values, init_data.columns.tolist()