Proper documentation for parameters

[ioana-blue]

Right now there are too many Number of epochs to train the model instead of the proper description for the parameters. It would be great to fix this --- I'm guessing this is one of the part that other users may fiddle with the most. Thank you!

[CoderPat]

Ye this one will be in the end of the backlog since I don't have too much time at the moment, but feel free to update it, would be appreciated.