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:08:41 +08:00
parent 959f31db33
commit 96eba1f850
2 changed files with 24 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

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