python - 在 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