Fit automated method instead of manually write it.

[thanoscs]

Well done for your excellent work!
I would like to ask you why don't you wrap the model with the parameters to the ".fit" function and you write it from the start. Is it because you want to observe additional informations for the training session or because ".fit" cannot be used in the current code?