YAO/YAO-optimizer
1
0
Fork 0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-12-13 08:26:43 +00:00
Code Issues Releases Wiki Activity

update

Browse Source
This commit is contained in:
Newnius
2020-05-02 01:34:47 +08:00
parent 05c0a7a90b
commit f3cef4f20a
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
serve.py
View File

@@ -14,7 +14,7 @@ import csv
class Config:
feature_columns = list(range(0, 1))
feature_columns = list(range(0, 2))
label_columns = [1]
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)
@@ -184,7 +184,7 @@ def train_models():
if train_X.shape[0] < 200:
config.batch_size = 16
train(config, train_X, train_Y, valid_X, valid_Y)
train(config, train_X[:, :1], train_Y, valid_X[:, :1], valid_Y)
lock.release()