Update evaluate.py

This commit is contained in:
jinzr 2024-05-21 15:13:06 +08:00
parent 3be889d896
commit a87f691b54

View File

@ -40,7 +40,7 @@ from at_datamodule import AudioSetATDatamodule
try: try:
from sklearn.metrics import average_precision_score from sklearn.metrics import average_precision_score
except: except:
raise ImportError(f"{ex}\nPlease run\n" "pip3 install -U scikit-learn") raise ImportError(f"Please run\n" "pip3 install -U scikit-learn")
from train import add_model_arguments, get_model, get_params, str2multihot from train import add_model_arguments, get_model, get_params, str2multihot
from icefall.checkpoint import ( from icefall.checkpoint import (