Fix metrics visualization v2 sample

2024-01-15 09:48:32 -03:00 · 2024-01-15 09:48:32 -03:00 · 158e259aad
parent 1cc31bb917
commit 158e259aad
1 changed files with 1 additions and 1 deletions
--- a/samples/test/metrics_visualization_v2.py
+++ b/samples/test/metrics_visualization_v2.py
@ -89,7 +89,7 @@ def wine_classification(metrics: Output[ClassificationMetrics]):
    y_predict = cross_val_predict(rfc, X_train, y_train, cv=3, method='predict')
    fpr, tpr, thresholds = roc_curve(
        y_true=y_train, y_score=y_scores[:, 1], pos_label=True)
-    metrics.log_roc_curve(fpr, tpr, thresholds)
+    metrics.log_roc_curve(fpr[1:], tpr[1:], thresholds[1:])


@component(