@@ -106,6 +106,9 @@ class _ModelMixin(abc.ABC):
if opts.optimizer == "rmsprop":
opt_kwargs["alpha"] = 0.9
+ if opts.optimizer in ["rmsprop", "adam"]:
+ opt_kwargs["eps"] = 1e-6
+
self.opt = optimizer(opts.optimizer,
self.clf,
opts.learning_rate,