Updated impurity_improvement to match newer declaration

EvgeniDubov / hellinger-distance-criterion

Random Forest model using Hellinger Distance as split criterion

BSD 3-Clause "New" or "Revised" License

31 stars 12 forks source link

Closed harish1996 closed 3 years ago

harish1996 commented 3 years ago

As mentioned in https://github.com/EvgeniDubov/hellinger-distance-criterion/issues/9#issue-853080199, sklearn's declaration has been updated since v0.24.1
Instead of calling children_impurity from the method, the method gets the right and left impurity as arguments.

EvgeniDubov commented 3 years ago

@harish1996 thanks a lot for the contribution