CaioUSC commited on
Commit
de68df4
·
verified ·
1 Parent(s): 68ab802

Update app/factory_predictor.py

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