Update app/factory_predictor.py

app/factory_predictor.py

@@ -2,11 +2,11 @@ import pandas as pd
 from sklearn.linear_model import LogisticRegression
 from sklearn.preprocessing import LabelEncoder
 from sklearn.metrics import accuracy_score
-from sklearn.tree import DecisionTreeRegressor
+from sklearn.tree import DecisionTreeClassifier
 
 class FactoryPredictor:
     def __init__(self):
-        self.model = DecisionTreeRegressor()
+        self.model = DecisionTreeClassifier(random_state=0)
         self.le_diagnosis = LabelEncoder()
         self.temperature = 0
         self.pressure = 0
@@ -30,7 +30,7 @@ class FactoryPredictor:
         data_test = pd.read_csv(csv_test)
 
         # Encode categorical variables
-        data_test['diagnosis'] = self.le_diagnosis.fit_transform(data_test['diagnosis'])
+        data_test['diagnosis'] = self.le_diagnosis.transform(data_test['diagnosis'])
 
         # Split features and target
         X_test = data_test[