Updated documentation on README

The README requires updates with respect to documentation - initialising models, the loss functions, etc. Expecting an extensive documentation present for each function and class present in the readthedocs documentation. A few things like performance metrics need to be added from the paper as well.