analist commited on
Commit
8e49d48
·
verified ·
1 Parent(s): 6cd1c2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -5,7 +5,7 @@ import matplotlib.pyplot as plt
5
  from sklearn.tree import plot_tree, export_text
6
  import seaborn as sns
7
  from sklearn.preprocessing import LabelEncoder
8
- import pickle
9
 
10
  def load_model_and_data():
11
  # Ici vous chargeriez votre modèle et données
@@ -14,8 +14,7 @@ def load_model_and_data():
14
  # X = loaded_X
15
  # y = loaded_y
16
  # feature_names = X.columns
17
- with open('DecisionTreeModel.pkl', 'rb') as f:
18
- model = pickle.load(f)
19
  data = pd.read_csv('exported_named_train.csv')
20
  X = data.drop("Target", axis=1)
21
  y = data['Target']
 
5
  from sklearn.tree import plot_tree, export_text
6
  import seaborn as sns
7
  from sklearn.preprocessing import LabelEncoder
8
+ import joblib
9
 
10
  def load_model_and_data():
11
  # Ici vous chargeriez votre modèle et données
 
14
  # X = loaded_X
15
  # y = loaded_y
16
  # feature_names = X.columns
17
+ model = joblib.load('DecisionTreeModel.joblib')
 
18
  data = pd.read_csv('exported_named_train.csv')
19
  X = data.drop("Target", axis=1)
20
  y = data['Target']