Can I send callbacks to a KerasClassifier?

后端 未结 3 722
清歌不尽
清歌不尽 2020-12-31 06:11

I want the classifier to run faster and stop early if the patience reaches the number I set. In the following code it does 10 iterations of fitting the model.



        
3条回答
  •  南笙
    南笙 (楼主)
    2020-12-31 06:59

    This is what I have done

    results = cross_val_score(estimator, X, Y, cv=kfold,
                          fit_params = {'callbacks': [checkpointer,plateau]})
    

    and has worked so far

提交回复
热议问题