Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 =
|
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}"
|