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 19:03:38 +08:00
parent 2613b50330
commit d1aa586432
2 changed files with 24 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@@ -8,7 +8,7 @@ from model.model_tensorflow import train, predict
class Config:
feature_columns = list(range(2, 9))
feature_columns = list(range(2, 3))
label_columns = [4, 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)