Skip to content
Snippets Groups Projects
Select Git revision
  • 73881adf6fe3bd3bdd329d6cb69c54c71f704fe8
  • master default
  • object
  • develop protected
  • private_algos
  • cuisine
  • SMOTE
  • revert-76c4cca5
  • archive protected
  • no_graphviz
  • 0.0.1
11 results

DecisionTree.py

Blame
  • DecisionTree.py 378 B
    from sklearn import tree
    # from sklearn.multiclass import OneVsRestClassifier
    from ModifiedMulticlass import OneVsRestClassifier
    
    # Add weights 
    
    def DecisionTree(data, labels, arg, weights):
        classifier = OneVsRestClassifier(tree.DecisionTreeClassifier(max_depth=arg))
        classifier.fit(data, labels, sample_weight=weights)
        return classifier, classifier.predict(data)