From d1aa58643244db3f34fabea533659618b7f638b8 Mon Sep 17 00:00:00 2001 From: Newnius Date: Wed, 29 Apr 2020 19:03:38 +0800 Subject: [PATCH] add files --- .idea/workspace.xml | 44 +++++++++++++++++++++++--------------------- main.py | 2 +- 2 files changed, 24 insertions(+), 22 deletions(-) diff --git a/.idea/workspace.xml b/.idea/workspace.xml index 40f7d01..73d3c30 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -1,7 +1,9 @@ - + + + - + - - + + @@ -57,11 +59,11 @@ - + - - + + @@ -96,8 +98,8 @@ @@ -151,7 +153,7 @@ - + @@ -182,12 +184,12 @@ - @@ -257,22 +259,22 @@ - + - - + + - + - + - - + + - + diff --git a/main.py b/main.py index 9d6b71b..49c351f 100644 --- a/main.py +++ b/main.py @@ -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)