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}
|