MUV-689任务时候出现问题

[YvoGao]

File "finetune.py", line 319, in main
fine_tune.train()
File "finetune.py", line 191, in train
self._test(model, test_loader)
File "finetune.py", line 311, in _test
self.roc_auc = roc_auc_score(labels, predictions[:,1])
File "/data/gaoyunlong/anaconda3/envs/pvl/lib/python3.8/site-packages/sklearn/utils/_param_validation.py", line 214, in wrapper
return func(*args, **kwargs)
File "/data/gaoyunlong/anaconda3/envs/pvl/lib/python3.8/site-packages/sklearn/metrics/_ranking.py", line 627, in roc_auc_score
return _average_binary_score(
File "/data/gaoyunlong/anaconda3/envs/pvl/lib/python3.8/site-packages/sklearn/metrics/_base.py", line 75, in _average_binary_score
return binary_metric(y_true, y_score, sample_weight=sample_weight)
File "/data/gaoyunlong/anaconda3/envs/pvl/lib/python3.8/site-packages/sklearn/metrics/_ranking.py", line 382, in _binary_roc_auc_score
raise ValueError(
ValueError: Only one class present in y_true. ROC AUC score is not defined in that case.