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 ab84173 commit a5f8ce8Copy full SHA for a5f8ce8
src/predictive_model/classification/classification.py
@@ -153,7 +153,15 @@ def _update(job: Job, data: DataFrame) -> dict:
153
if not x.empty:
154
y = x['label']
155
156
- models[cluster].partial_fit(x.drop('label', 1), y.values.ravel())
+ try:
157
+ models[cluster].partial_fit(x.drop('label', 1), y.values.ravel())
158
+ except (NotImplementedError, KeyError):
159
160
+ models[cluster].partial_fit(x.drop('label', 1).T, y.values.ravel())
161
+ except KeyError:
162
+ models[cluster].partial_fit(x.drop('label', 1).values, y.values.ravel())
163
+ except Exception as exception:
164
+ raise exception
165
166
return {ModelType.CLUSTERER.value: clusterer, ModelType.CLASSIFIER.value: models}
167
0 commit comments