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

add files

Browse Source
This commit is contained in:
Newnius
2020-04-29 22:29:32 +08:00
parent 43210efed9
commit 5507bd66a8
3 changed files with 178 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
main.py
View File

@@ -4,9 +4,11 @@ import os
from sklearn.model_selection import train_test_split
from model_tensorflow import train, predict
frame = "tensorflow"
class Config:
feature_columns = list([2,5])
feature_columns = list([2, 5])
label_columns = [5]
feature_and_label_columns = feature_columns + label_columns
label_in_feature_columns = (lambda x, y: [x.index(i) for i in y])(feature_columns, label_columns)
@@ -51,9 +53,9 @@ class Config:
if not os.path.exists(figure_save_path):
os.mkdir(figure_save_path)
#used_frame = frame
#model_postfix = {"pytorch": ".pth", "keras": ".h5", "tensorflow": ".ckpt"}
#model_name = "model_" + continue_flag + used_frame + model_postfix[used_frame]
used_frame = frame
model_postfix = {"pytorch": ".pth", "keras": ".h5", "tensorflow": ".ckpt"}
model_name = "model_" + continue_flag + used_frame + model_postfix[used_frame]
class Data: