Fixed error by passing train df in the model
Merge pull request #68 from prathmesh/master

Fixed error by passing train df in the model