用于多类分类器的网格搜索cv

时间:2017-08-21 16:02:23

标签: python grid-search multiclass-classification

我正在使用ANN作为多类分类器来解决目标变量可以取3个值的问题:-1,0,1。

我使用OneHotEncoder使用以下代码转换输出变量 -

labelencoder_y = LabelEncoder()
y = labelencoder_y.fit_transform(y) 
uniques, ids = np.unique(y, return_inverse=True)

现在采用以下格式 -

array([[0, 0, 1],[1, 0, 0],[0, 1, 0]....

预测看起来像 -

array([[0, 0.3, 0.7],[0.6, 0.3, 0.1],[0.1, 0.5, 0.4]...

我想使用网格搜索cv来提高我的准确率,但我很难做到这一点。我总是得到一个错误。这是我正在使用的代码 -

def build_classifier():
    classifier = Sequential()
    classifier.add(Dense(activation = "relu", input_dim = 28, units = 32, use_bias=True, kernel_initializer= 'normal'))
    classifier.add(Dense(activation="softmax", units = 3, kernel_initializer = 'normal'))
    classifier.compile(optimizer= 'adam', loss= 'categorical_crossentropy', metrics= ['accuracy'])
    return classifier

classifier = KerasClassifier(build_fn = build_classifier)
parameters = {'batch_size': [25,32],
              'epochs' : [50]
              }

grid_search = GridSearchCV(estimator= classifier, param_grid= parameters, scoring= 'f1_micro', cv=10,return_train_score = True)
grid_search = grid_search.fit(X_train,y_train)

我收到以下错误 -

ValueError: Can't handle mix of multilabel-indicator and multiclass

我知道问题是GridSearchCV(...)中的评分参数。我找不到合适的参数。我尝试用这个来创建自己的损失函数:

def my_check(y_true, y_pred):
    y_true = np.take(uniques,np.argmax(y_true,1))
    y_pred = np.take(uniques,np.argmax(y_pred,1))
    accuracy = accuracy_score(y_true, y_pred)
    return accuracy 

score = make_scorer(my_check, greater_is_better=True)

但我再次收到错误:

ValueError: axis(=1) out of bounds

知道我该怎么办?

0 个答案:

没有答案