Question

我想计算机器学习模型处理数据分类所花费的时间。我正在使用scikit-learn。我可以通过使用

来计算准确性，查全率和准确性。

function solution(A){
    let multi = 1;
    for(i = 1; i < A.length; i++){
        multi *= A[i]
    } 

    if(multi = 30){
        return 1
    } else if (multi = -30){
        return -1
    } else if (multi = 0){
        return 0
    } else{
        console.log("hey hey");
    }
}

solution(A = [1,2,0,-5])

但是我不知道如何计算分类的评估时间。有

Answer 1

赞：

from time import time

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

Answer 2

您可以简单地使用以下内容：

import time
.....
start = time.time()
y_pred = model.predict(Xtest)
end = time.time()
eval_time = end-start # in seconds

只需用start = ...和end = ...包装要检查的部分代码

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

2 个答案: