We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4505d10 commit bc71e73Copy full SHA for bc71e73
pureskillgg_dsdk/ds_models/s3_scikit.py
@@ -34,6 +34,9 @@ def _use_model(self, dataframe):
34
if self._model_type == "MiniBatchKMeans":
35
labels = self._loaded_model.predict(dataframe)
36
return labels
37
+ if self._model_type == "SGDClassifier":
38
+ labels = self._loaded_model.predict_proba(dataframe)
39
+ return labels
40
raise Exception(f"Unknown model_type {self._model_type}")
41
42
def invoke(self, dataframe):
0 commit comments