fix model input features (#198)

Co-authored-by: Eugene Fedorenko <[email protected]>
microsoft · Feb 20, 2020 · 07da4ef · 07da4ef
1 parent 171a27f
commit 07da4ef
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/diabetes_regression/scoring/score.py b/diabetes_regression/scoring/score.py
@@ -48,8 +48,8 @@ def init():
 
 
 input_sample = numpy.array([
-    [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
-    [10, 9, 8, 7, 6, 5, 4, 3, 2, 1]])
+    [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0],
+    [10.0, 9.0, 8.0, 7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0]])
 output_sample = numpy.array([
     5021.509689995557,
     3693.645386402646])

diff --git a/diabetes_regression/training/train.py b/diabetes_regression/training/train.py
@@ -95,8 +95,8 @@ def main():
     dataset = run.input_datasets['training_data']
     if (dataset):
         df = dataset.to_pandas_dataframe()
-        X = df.values
-        y = df.Y
+        X = df.drop('Y', axis=1).values
+        y = df['Y'].values
     else:
         e = ("No dataset provided")
         print(e)