From c9bc9aa8cc8fdb6769aac0b9ee81ee938225d34f Mon Sep 17 00:00:00 2001
From: guipleite <guipleite@gmail.com>
Date: Wed, 31 Mar 2021 14:46:24 -0300
Subject: [PATCH] fixing deploy

---
 dvc.lock             | 24 ++++++++++++------------
 requirements.txt     |  2 +-
 results/metrics.json |  2 +-
 src/train.py         |  6 +-----
 4 files changed, 15 insertions(+), 19 deletions(-)

diff --git a/dvc.lock b/dvc.lock
index 904855e..fb617a6 100644
--- a/dvc.lock
+++ b/dvc.lock
@@ -26,12 +26,12 @@ stages:
       md5: 260904955bdf53e03a72aa2a45fa0297
       size: 4451
     - path: ./src/train.py
-      md5: 42f68d29be9da054ca6e6f0e5c142778
-      size: 1094
+      md5: 1b5c6c1786d40c9505b2261f11a3b274
+      size: 1002
     outs:
     - path: ./models/model.joblib
-      md5: 287cf6a0a3fabe1103cfaca91bbc9384
-      size: 17122152
+      md5: 6e7186e0d9e5026be46572e2cb02ca06
+      size: 16869560
   evaluate:
     cmd: python3 ./src/evaluate.py ./data/weatherAUS_processed.csv ./src/model.py
       ./models/model.joblib
@@ -40,8 +40,8 @@ stages:
       md5: 59e89e62fb8f9face4901630d1de3e16
       size: 19507550
     - path: ./models/model.joblib
-      md5: 287cf6a0a3fabe1103cfaca91bbc9384
-      size: 17122152
+      md5: 6e7186e0d9e5026be46572e2cb02ca06
+      size: 16869560
     - path: ./src/evaluate.py
       md5: 7e466368d793d09316fc1e078111a9de
       size: 882
@@ -50,13 +50,13 @@ stages:
       size: 4451
     outs:
     - path: ./results/metrics.json
-      md5: 1e190be0481c320f15ff811a21e32490
-      size: 121
+      md5: af950439e97764b5bf7f91322f6aa8bf
+      size: 120
     - path: ./results/precision_recall_curve.png
-      md5: d4b1401d5c8601c2f3934e8028785302
-      size: 17033
+      md5: 9b817eb824b73c484bde8060fa01507a
+      size: 17106
     - path: ./results/roc_curve.png
-      md5: c38ee7702a4749a5ff60c94a9e575a50
-      size: 20026
+      md5: 7530a23497d03b976795542f5dd4762f
+      size: 19956
   std_check:
     cmd: src/scripts/Scripts/std_check.sh ./
diff --git a/requirements.txt b/requirements.txt
index ca093ed..6f7a3c9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-scikit-learn==0.21.3
+scikit-learn>=0.23
 pandas
 seaborn
 matplotlib
diff --git a/results/metrics.json b/results/metrics.json
index 2a16b2d..521c160 100644
--- a/results/metrics.json
+++ b/results/metrics.json
@@ -1 +1 @@
-{"accuracy": 0.8469057178457704, "recall": 0.9647707074297902, "precision": 0.8567414843577359, "f1": 0.9075526276188406}
\ No newline at end of file
+{"accuracy": 0.8464349993077669, "recall": 0.9659438322076075, "precision": 0.8555415617128463, "f1": 0.907396894306228}
\ No newline at end of file
diff --git a/src/train.py b/src/train.py
index cca3f6b..e88963b 100644
--- a/src/train.py
+++ b/src/train.py
@@ -3,19 +3,15 @@
 import pickle
 import os
 import json
+import joblib
 
 # import sklearn.external.joblib as extjoblib
-import joblib
 
 DATA_PATH = os.path.abspath(sys.argv[1])
 # PROJ_PATH = os.path.abspath(sys.argv[2])
-# MODEL_PATH = PROJ_PATH+"/src/model.py"
 MODEL_PATH = sys.argv[2]
 PARAM = int(sys.argv[3])
 
-# if sys.argv[4]:
-#     OUTPUT_PATH = sys.argv[4]
-
 sys.path.insert(1, MODEL_PATH)