1
0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-12-13 00:16:44 +00:00
This commit is contained in:
2020-05-01 17:21:35 +08:00
parent 96d4ecc483
commit 7fc3ba4732
2 changed files with 32 additions and 30 deletions

View File

@@ -16,8 +16,8 @@ from io import StringIO
class Config:
feature_columns = list([2, 5])
label_columns = [5]
feature_columns = list(range(0, 8))
label_columns = [5, 6, 7]
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)