2016-11-30 6 views
2

私はscikitlearnを使用してSVMをトレーニングしています。私は、バリデーションセットで現在のモデルの精度をテストするために、たびにトレーニングを一時停止することが可能であると思っていました。最終的には、検証確度曲線を生成したいと考えています。 .Fit()を使用するとSVMが完全に練習されますが、最後に1つの精度データポイントが与えられます。scikit learn SVM停止中にトレーニング中にアクセスする

答えて

0

実際に使用できる小さなトリックがあります。

SVC分類子のパラメータmax_iterで再生できます。 たとえば、反復回数が異なる複数の分類子を取得できます。ここで

は、あなたが何ができるかです:そう

import numpy as np 

for i in np.arange(10, 1000, 100): 
    svm = SVC(max_iter=i) # and your other parameters 
    svm.fit(X, y) 
    ... # here retrieve your metrics 

は、分類器が訓練の異なるレベルで実行する方法を紹介します。

+1

私はStack Overflowに投稿する前にこれを実際に試しましたが、.Fit()という名前のどこかで読んだことを思い出しました。私は.Fit()がループのたびに中断したところからピックアップしたいと思うでしょう。 –

関連する問題