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 34a30c7 commit d47faceCopy full SHA for d47face
sklbench/benchmarks/sklearn_estimator.py
@@ -191,19 +191,6 @@ def get_subset_metrics_of_estimator(
191
}
192
)
193
elif task == "clustering":
194
- if hasattr(estimator_instance, "inertia_"):
195
- # compute inertia manually using distances to cluster centers
196
- # provided by KMeans.transform
197
- metrics.update(
198
- {
199
- "inertia": float(
200
- np.power(
201
- convert_to_numpy(estimator_instance.transform(x)).min(axis=1),
202
- 2,
203
- ).sum()
204
- )
205
- }
206
207
if hasattr(estimator_instance, "predict"):
208
y_pred = convert_to_numpy(estimator_instance.predict(x))
209
metrics.update(
0 commit comments