在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