【发布时间】:2020-08-13 11:04:42
【问题描述】:
我正在尝试使用 tf keras 进行多类分类。我总共有 20 个标签,我拥有的总数据是 63952并且我尝试了以下代码
features = features.astype(float)
labels = df_test["label"].values
encoder = LabelEncoder()
encoder.fit(labels)
encoded_Y = encoder.transform(labels)
dummy_y = np_utils.to_categorical(encoded_Y)
然后
def baseline_model():
model = Sequential()
model.add(Dense(50, input_dim=3, activation='relu'))
model.add(Dense(40, activation='softmax'))
model.add(Dense(30, activation='softmax'))
model.add(Dense(20, activation='softmax'))
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
return model
终于
history = model.fit(data,dummy_y,
epochs=5000,
batch_size=50,
validation_split=0.3,
shuffle=True,
callbacks=[ch]).history
我对此的准确性很差。我该如何改进呢?
【问题讨论】:
标签: machine-learning keras multiclass-classification