matsammut commited on
Commit
69eedcf
·
verified ·
1 Parent(s): 5c54263

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ def predict_hb(age, workclass, education, occupation, race, gender, capital_ga
70
  # ann_model = pickle.load(ann_model_file)
71
  scaler = StandardScaler()
72
  X = scaler.fit_transform(fixed_features)
73
- hb_model = pickle.load(open('hdbscan_model.pkl', 'rb'))
74
  prediction = hdbscan.approximate_predict(hb_model,fixed_features)
75
  # prediction = 1
76
  return f"Predicted Cluster (HDBSCAN): {prediction}"
 
70
  # ann_model = pickle.load(ann_model_file)
71
  scaler = StandardScaler()
72
  X = scaler.fit_transform(fixed_features)
73
+ hb_model = joblib.load('hdbscan_model.joblib')
74
  prediction = hdbscan.approximate_predict(hb_model,fixed_features)
75
  # prediction = 1
76
  return f"Predicted Cluster (HDBSCAN): {prediction}"