How to define Learning Rates and Early_STOP

[luoda888]

No description provided.

[shenweichen]

Hi,@luoda888
You can use any models in DeepCTR like a keras model object.
Here is a example of how to set learning rate and early_stop:

import deepctr
from tensorflow.python.keras.optimizers import Adam,Adagrad
from tensorflow.python.keras.callbacks import EarlyStopping

model = deepctr.models.DeepFM({"sparse": sparse_feature_dict, "dense": dense_feature_list})
model.compile(Adagrad('0.0808'),'binary_crossentropy',metrics=['binary_crossentropy'])

es = EarlyStopping(monitor='val_binary_crossentropy')
history = model.fit(model_input, data[target].values,batch_size=256, epochs=10, verbose=2, validation_split=0.2,callbacks=[es] )

[luoda888]

OK! Thanks