search

ankonzoid / LearningX

Deep & Classical Reinforcement Learning + Machine Learning Examples in Python
MIT License
349 stars 187 forks source link

Add Sklearn compatibility to ModelTree. #4

Closed gw0 closed 6 years ago

gw0 commented 6 years ago

Added three methods to ModelTree to make it more compatible with scikit-learn. Now you can use all cross-validation functions from scikit-learn.