TODOs and sklearn style comments for methods
predict returns an np.array
Less generic error message when not fitted in prediction methods