From 959f31db33fc036bc7d8ed46eb46452d49e2d19a Mon Sep 17 00:00:00 2001 From: Newnius Date: Wed, 29 Apr 2020 19:05:37 +0800 Subject: [PATCH] add files --- .idea/workspace.xml | 2 +- main.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.idea/workspace.xml b/.idea/workspace.xml index 73d3c30..0740440 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -153,7 +153,7 @@ - + diff --git a/main.py b/main.py index 49c351f..77372f1 100644 --- a/main.py +++ b/main.py @@ -8,8 +8,8 @@ from model.model_tensorflow import train, predict class Config: - feature_columns = list(range(2, 3)) - label_columns = [4, 5] + feature_columns = list(range(2)) + label_columns = [4] 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)