Convert sklearn.svm SVC classifier to Keras implementation

前端 未结 2 1077
难免孤独
难免孤独 2021-01-01 19:35

I\'m trying to convert some old code from using sklearn to Keras implementation. Since it is crucial to maintain the same way of operation, I want to understand if I\'m doin

2条回答
  •  被撕碎了的回忆
    2021-01-01 20:36

    If you are making a classifier, you need squared_hinge and regularizer, to get the complete SVM loss function as can be seen here. So you will also need to break your last layer to add regularization parameter before performing activation, I have added the code here.

    These changes should give you the output

    from keras.regularizers import l2
    from keras.models import Sequential
    from keras.layers import Dense
    
    model = Sequential()
    model.add(Dense(64, activation='relu'))
    model.add(Dense(1), W_regularizer=l2(0.01))
    model.add(activation('softmax'))
    model.compile(loss='squared_hinge',
                  optimizer='adadelta',
                  metrics=['accuracy'])
    model.fit(X, Y_labels)
    
    

    Also hinge is implemented in keras for binary classification, so if you are working on a binary classification model, use the code below.

    from keras.regularizers import l2
    from keras.models import Sequential
    from keras.layers import Dense
    
    model = Sequential()
    model.add(Dense(64, activation='relu'))
    model.add(Dense(1), W_regularizer=l2(0.01))
    model.add(activation('linear'))
    model.compile(loss='hinge',
                  optimizer='adadelta',
                  metrics=['accuracy'])
    model.fit(X, Y_labels)
    
    

    If you cannot understand the article or have issues with the code, feel free to comment. I had this same issue a while back, and this GitHub thread helped me understand, maybe go through it too, some of the ideas here are directly from here https://github.com/keras-team/keras/issues/2588

提交回复
热议问题