Fix catboost version

akarazeev · Oct 22, 2018 · 3fad51f · 3fad51f
1 parent 9186fb7
commit 3fad51f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/bot.py b/bot.py
@@ -129,9 +129,9 @@ def text_handler(bot, update):
             'row1': [data['month'], data['day'], data['temp'], data['RH'], data['speed'], data['rain']]
         }
         sample = pd.DataFrame.from_dict(sample, orient='index')
-        print(sample)
-        # prediction = model.predict(sample)
-        prediction = [4.0]
+        # print(sample)
+        prediction = model.predict(sample)
+        # prediction = [4.0]
 
         data['firearea'] = round(prediction[0])
         data['nearest'] = round(min_distance(float(location['latitude']), float(location['longitude']))[0])

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 asn1crypto==0.24.0
-catboost==0.10.3
+catboost==0.4.1
 certifi==2018.10.15
 cffi==1.11.5
 chardet==3.0.4