Great catch 付友 — thank you!

I’ve removed the “-” from Gini impurity but I have left it for entropy as this is the formula from Scikit Learn and Spark docs. The links and info I’m referring to are below:

Scikit-learn doc link

Scikit-learn explanation

Spark doc explanation

Thanks again,

Stacey