获取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