sklearn: How to reset a Regressor or classifier object in sknn

后端 未结 2 1583
借酒劲吻你
借酒劲吻你 2021-02-20 14:43

I have defined a regressor as follows:

nn1 = Regressor(
layers=[
    Layer(\"Rectifier\", units=150),
    Layer(\"Rectifier\", units=100),
    Layer(\"Linear\")]         


        
2条回答
  •  梦如初夏
    2021-02-20 15:10

    The pattern that I use for cross validation instantiates a new classifier for each training/test pair:

    from sklearn.cross_validation import KFold
    
    kf = KFold(len(labels),n_folds=5, shuffle=True)
    for train, test in kf:
        clf = YourClassifierClass()
        clf.fit(data[train],labels[train])
        # Do evaluation with data[test] and labels[test]
    

    You can save your current best classifier in a separate variable and access its parameters after cross validation (this is also useful if you want to try different parameters).

提交回复
热议问题