Fix #890: Add Jaime's model to AimodelSS, AimodelDataFactory

oceanprotocol · Apr 30, 2024 · 780f7c3 · 780f7c3
1 parent 9e87b23
commit 780f7c3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pdr_backend/aimodel/aimodel_factory.py b/pdr_backend/aimodel/aimodel_factory.py
@@ -93,9 +93,11 @@ def build(
         if do_constant or ss.calibrate_probs == "None":
             pass
         elif ss.calibrate_probs == "CalibratedClassifierCV_5x":
+            N = X.shape[0]
+            method = "sigmoid" if N < 200 else "isotonic"
             cv = min(smallest_n, 5)
             if cv > 1:
-                skm = CalibratedClassifierCV(skm, cv=cv)
+                skm = CalibratedClassifierCV(skm, method=method, cv=cv)
         else:
             raise ValueError(ss.calibrate_probs)