ankonzoid / LearningX

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

Showing function in leaf nodes when using linear regression #10

Open vildursn opened 4 years ago

vildursn commented 4 years ago

It's useful to see the functions the tree has fitted when using linear regression, so I added functionality for this. For now, all the other models get_params() functions just returns None, but based on what is useful information this should be changed for them as well.

Also fixed a small bug in the logistic_regression file.