python - 如何计算机器学习模型进行分类的时间？

Question

我想计算机器学习模型处理数据分类所需的时间。我正在使用 scikit-learn。我可以使用以下方法计算准确度、召回率和精度：

from sklearn.metrics import classification_report
from sklearn.metrics import confusion_matrix
from sklearn.metrics import accuracy_score
.....
model.fit(Xtrain, Ytrain)
y_pred = model.predict(Xtest)
print(accuracy_score(Ytest, y_pred))
print(confusion_matrix(Ytest, y_pred))
print(classification_report(Ytest, y_pred))

但我不知道如何计算分类的评估时间。在那儿

标签： pythonscikit-learn

Answer 1

像这样：

from time import time

t0 = time()
fit_classifier()
print(time() - t0)