diff --git a/server/nightr/strategies/svm_strat.py b/server/nightr/strategies/svm_strat.py index 90fa84f..789aa8c 100644 --- a/server/nightr/strategies/svm_strat.py +++ b/server/nightr/strategies/svm_strat.py @@ -54,5 +54,5 @@ def perform_svm_pred(context: Context) -> Prediction: p.reasons.append("Our only two data points have 11 dimensions") p.reasons.append("We are using a SVM") - p.probability = predict(X) + p.probability = float(predict(X)) return p