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:20:31 +08:00
parent e41f541f3b
commit 43210efed9
2 changed files with 80 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
main.py
View File

@@ -2,10 +2,9 @@ import pandas as pd
import numpy as np
import os
from sklearn.model_selection import train_test_split
frame = "tensorflow"
from model_tensorflow import train, predict
class Config:
feature_columns = list([2,5])
label_columns = [5]
@@ -52,9 +51,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: