lg-lima1 commited on
Commit
7eda480
·
1 Parent(s): 8718717

feat: change predictor model to RandomForestClassifier

Browse files
Files changed (1) hide show
  1. app/factory_predictor.py +2 -2
app/factory_predictor.py CHANGED
@@ -1,11 +1,11 @@
1
  import pandas as pd
2
- from sklearn.linear_model import LogisticRegression
3
  from sklearn.preprocessing import LabelEncoder
4
  from sklearn.metrics import accuracy_score
5
 
6
  class FactoryPredictor:
7
  def __init__(self):
8
- self.model = LogisticRegression()
9
  self.le_diagnosis = LabelEncoder()
10
  self.temperature = 0
11
  self.pressure = 0
 
1
  import pandas as pd
2
+ from sklearn.ensemble import RandomForestClassifier
3
  from sklearn.preprocessing import LabelEncoder
4
  from sklearn.metrics import accuracy_score
5
 
6
  class FactoryPredictor:
7
  def __init__(self):
8
+ self.model = RandomForestClassifier()
9
  self.le_diagnosis = LabelEncoder()
10
  self.temperature = 0
11
  self.pressure = 0