doubleplusplus / incremental_decision_tree-CART-Random_Forest

incremental CART decision tree, based on the hoeffding tree i.e. very fast decision tree (VFDT), which is proposed in this paper "Mining High-Speed Data Streams" by Domingos & Hulten (2000). And a newly extended model "Extremely Fast Decision Tree" (EFDT) by Manapragada, Webb & Salehi (2018). Added new implementation of Random Forest
100 stars 28 forks source link

update and predict now take sklearn-style arrays (+ PEP8) #2

Closed Adrien-Luxey closed 5 years ago

Adrien-Luxey commented 5 years ago

Hi!

I've been using your VFDT code, thank you a lot, it's very clear and well-made.

I'm using VFDT in a pipeline where I sometimes want to pass it arrays to update/predict at once, so:

In case you are interested, have this pull request, with regards, Adrien Luxey

PS: If you want to be even more sklearn compatible, you should just rename your update into fit. You could also add a fit_predict function doing both, as in e.g. KMeans.