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

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