Skip to content
Snippets Groups Projects
Select Git revision
  • d0872e1e6ed08c27a6a4d5fee340e44a585c0810
  • main default protected
2 results

plot_prec_rec.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)