sklearn-porter
sklearn-porter copied to clipboard
Feature Request: Multi-label DecisionTreeClassifier
I have trained a multi-label DecisionTreeClassifier and when I ported the result is the following:
public static int predict(double[] features) {
int[] classes = new int[2];
if (features[11] <= 12.5) {
if (features[10] <= 182.5) {
if (features[12] <= 72.5) {
if (features[13] <= 63.0) {
if (features[7] <= 767.5) {
classes[0] = 20;
classes[1] = 5;
// Here the result shoud be:
//classes[0][0] = 20; classes[0][1] = 5; classes[1][0] = 25; classes[1][1] = 0; And so on...
}else{
//Huge amount of ifs
}
}
}
}
}
The full decision tree is here:
I really appreciate this feature in your porter. =]
Additionally, if this feature is already present in the code, I haven't figured out how to use it.
Thank you.