Spaces:
Sleeping
Sleeping
Clement Vachet
commited on
Commit
·
e0f980a
1
Parent(s):
e9c74b2
style: use double quote
Browse files
classification/classifier.py
CHANGED
|
@@ -41,7 +41,7 @@ class Classifier:
|
|
| 41 |
model_path = os.path.join(parent_dir, "models", "model.pkl")
|
| 42 |
model = joblib.load(model_path)
|
| 43 |
|
| 44 |
-
features = np.array(data[
|
| 45 |
|
| 46 |
if features.shape[-1] != 4:
|
| 47 |
raise ValueError("Expected 4 features per input.")
|
|
@@ -51,7 +51,7 @@ class Classifier:
|
|
| 51 |
probabilities = model.predict_proba(features).tolist()
|
| 52 |
|
| 53 |
# Map predictions to class labels
|
| 54 |
-
iris_types = {0:
|
| 55 |
prediction_labels = [iris_types[pred] for pred in predictions]
|
| 56 |
|
| 57 |
return {"predictions": prediction_labels, "probabilities": probabilities}
|
|
|
|
| 41 |
model_path = os.path.join(parent_dir, "models", "model.pkl")
|
| 42 |
model = joblib.load(model_path)
|
| 43 |
|
| 44 |
+
features = np.array(data["features"])
|
| 45 |
|
| 46 |
if features.shape[-1] != 4:
|
| 47 |
raise ValueError("Expected 4 features per input.")
|
|
|
|
| 51 |
probabilities = model.predict_proba(features).tolist()
|
| 52 |
|
| 53 |
# Map predictions to class labels
|
| 54 |
+
iris_types = {0: "setosa", 1: "versicolor", 2: "virginica"}
|
| 55 |
prediction_labels = [iris_types[pred] for pred in predictions]
|
| 56 |
|
| 57 |
return {"predictions": prediction_labels, "probabilities": probabilities}
|