小编Jie*_* HE的帖子

我们知道我们可以在二元分类模型中的不平衡数据的拟合方法中传递一个类权重字典.我的问题是,当在输出层中仅使用1个节点进行sigmoid激活时,我们是否仍然可以在训练期间应用类权重？

model = Sequential()
model.add(Dense(64, activation='tanh',input_shape=(len(x_train[0]),)))
model.add(Dense(1, activation='sigmoid')) 

model.compile(
    optimizer=optimizer, 
    loss=loss, 
    metrics=metrics)

model.fit(
    x_train, y_train, 
    epochs=args.e, 
    batch_size=batch_size,
    class_weight={0: 1, 1: 3})