Skip to content

Commit b5cf1f7

Browse files
committed
Fix pylint error
1 parent cf96e82 commit b5cf1f7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cinema/classification/train.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,7 @@ def binary_classification_metrics(
219219
else:
220220
metrics["mcc"] = 0.0
221221
metrics["roc_auc"] = 0.0
222-
logger.info("Accuracy (%) | ROC AUC (%) | F1 (%) | MCC (%) | Specificity (%) | Sensitivity (%)")
222+
logger.info("Accuracy | ROC AUC | F1 | MCC | Specificity | Sensitivity")
223223
logger.info(
224224
f"{metrics['accuracy'] * 100:.2f} | {metrics['roc_auc'] * 100:.2f} | "
225225
f"{metrics['f1'] * 100:.2f} | {metrics['mcc'] * 100:.2f} | "
@@ -262,7 +262,7 @@ def multiclass_classification_metrics(
262262
else:
263263
metrics["mcc"] = 0.0
264264
metrics["roc_auc"] = 0.0
265-
logger.info("Accuracy (%) | ROC AUC (%) | F1 (%) | MCC (%)")
265+
logger.info("Accuracy | ROC AUC | F1 | MCC")
266266
logger.info(
267267
f"{metrics['accuracy'] * 100:.2f} | {metrics['roc_auc'] * 100:.2f} | "
268268
f"{metrics['f1'] * 100:.2f} | {metrics['mcc'] * 100:.2f}"

0 commit comments

Comments
 (0)