How to train multiclass documents data?

[shashankg7]

Hi,

Thanks for releasing the code in keras, the other implementations are in torch and tensorflow and theano which currently I am not familiar with.

I have one question though, how can I train multiclass documents with this model? Currently it seems to support only binary class data?

[alexander-rakhlin]

Thanks for the interest. I suggest softmax in last layers and categorical_crossentropy as loss:

model.add(Dense(num_classes))
model.add(Activation("softmax"))
model.compile(loss="categorical_crossentropy", optimizer="rmsprop", metrics=["accuracy"])

[worksduke]

来信收悉。