Clement Vachet commited on
Commit
e0f980a
·
1 Parent(s): e9c74b2

style: use double quote

Browse files
Files changed (1) hide show
  1. classification/classifier.py +2 -2
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['features'])
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: 'setosa', 1: 'versicolor', 2: 'virginica'}
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}