在 scikit 中获取训练时间

Question

我正在使用 scikit 来训练分类器。我想知道是否有一个选项可以获取分类器/估计器完成训练任务所需的时间。

Answer 1

只需使用Python的time模块即可。例如：

import time
from sklearn.neural_network import MLPClassifier
from sklearn.datasets import load_iris


model = MLPClassifier()
X, y = load_iris(return_X_y=True)
start = time.time()
model.fit(X, y)
stop = time.time()
print(f"Training time: {stop - start}s")
# prints: Training time: 0.20307230949401855s