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 18:42:01 +08:00
parent 34f2824db8
commit 97c839d344
2 changed files with 52 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
main.py
View File

@@ -4,14 +4,7 @@ import os
from sklearn.model_selection import train_test_split
frame = "tensorflow"
if frame == "pytorch":
from model.model_pytorch import train, predict
elif frame == "keras":
from model.model_keras import train, predict
elif frame == "tensorflow":
from model.model_tensorflow import train, predict
else:
raise Exception("Wrong frame seletion")
from model.model_tensorflow import train, predict
class Config:
@@ -143,6 +136,9 @@ def draw(config, origin_data, predict_norm_data):
predict_data = predict_norm_data * origin_data.std[config.label_in_feature_columns] + \
origin_data.mean[config.label_in_feature_columns]
print(label_data)
print(predict_data)
'''
for i in range(label_column_num):
plt.figure(i + 1)