Python users are more used to models having a `.fit()`, `.predict()` etc. I could create a PR with this feature if you would accept it?