caspervk/nightr
1
0
Fork 0
Code Releases 1 Activity

pls

Browse source
This commit is contained in:
Alexander Munch-Hansen 2019-04-07 00:03:50 +02:00
parent 65a49dfb95
commit fa63cb9e06
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
server/nightr/strategies/svm_strat.py
View file

@ -1,3 +1,5 @@
from pathlib import Path
from sklearn import svm from sklearn import svm
from sklearn.externals import joblib from sklearn.externals import joblib
import requests import requests
@ -36,7 +38,7 @@ def train():
joblib.dump(classifier, "nightness_classifier.pkl") joblib.dump(classifier, "nightness_classifier.pkl")
def predict(X): def predict(X):
classifier = joblib.load("nightness_classifier.pkl") classifier = joblib.load(str(Path(__file__).parent.joinpath("nightness_classifier.pkl")))
prob = classifier.predict_proba(np.array(X).reshape(1, -1)) prob = classifier.predict_proba(np.array(X).reshape(1, -1))
return prob[0, 1] return prob[0, 1]