Ignore:
Timestamp:
03/02/12 15:55:49 (2 years ago)
Author:
anzeh <anze.staric@…>
Branch:
default
Message:

Refactored AUC.

File:
1 edited

Legend:

Unmodified
Added
Removed
  • Orange/testing/unit/tests/test_evaluation_scoring.py

    r10425 r10426  
    4545        ds = data.Table("iris") 
    4646        test_results = testing.cross_validation([self.learner], ds, folds=5) 
    47         auc = scoring.AUC.by_pairs(test_results) 
     47        auc = scoring.AUC(test_results, multiclass=scoring.AUC.ByPairs) 
    4848 
    4949        self.assertEqual(len(auc), 1) 
     
    5252        ds = data.Table("iris") 
    5353        test_results = testing.cross_validation([self.learner], ds, folds=5) 
    54         auc = scoring.AUC.by_weighted_pairs(test_results) 
     54        auc = scoring.AUC(test_results, multiclass=scoring.AUC.ByWeightedPairs) 
    5555 
    5656        self.assertEqual(len(auc), 1) 
     
    5959        ds = data.Table("iris") 
    6060        test_results = testing.cross_validation([self.learner], ds, folds=5) 
    61         auc = scoring.AUC.one_against_all(test_results) 
     61        auc = scoring.AUC(test_results, multiclass=scoring.AUC.OneAgainstAll) 
    6262 
    6363        self.assertEqual(len(auc), 1) 
     
    6666        ds = data.Table("iris") 
    6767        test_results = testing.cross_validation([self.learner], ds, folds=5) 
    68         auc = scoring.AUC.weighted_one_against_all(test_results) 
     68        auc = scoring.AUC(test_results, multiclass=scoring.AUC.WeightedOneAgainstAll) 
    6969 
    7070        self.assertEqual(len(auc), 1) 
     
    7373        ds = data.Table("iris") 
    7474        test_results = testing.cross_validation([self.learner], ds, folds=5) 
    75         auc = scoring.AUC.single_class(test_results) 
    76         self.assertEqual(len(auc), 1) 
    77         auc = scoring.AUC.single_class(test_results, 0) 
    78         self.assertEqual(len(auc), 1) 
    79         auc = scoring.AUC.single_class(test_results, 1) 
    80         self.assertEqual(len(auc), 1) 
    81         auc = scoring.AUC.single_class(test_results, 2) 
     75        auc = scoring.AUC_for_single_class(test_results) 
     76        self.assertEqual(len(auc), 1) 
     77        auc = scoring.AUC_for_single_class(test_results, 0) 
     78        self.assertEqual(len(auc), 1) 
     79        auc = scoring.AUC_for_single_class(test_results, 1) 
     80        self.assertEqual(len(auc), 1) 
     81        auc = scoring.AUC_for_single_class(test_results, 2) 
    8282        self.assertEqual(len(auc), 1) 
    8383 
     
    8585        ds = data.Table("iris") 
    8686        test_results = testing.cross_validation([self.learner], ds, folds=5) 
    87         auc = scoring.AUC.pair(test_results, 0, 1) 
    88         self.assertEqual(len(auc), 1) 
    89         auc = scoring.AUC.pair(test_results, 0, 2) 
    90         self.assertEqual(len(auc), 1) 
    91         auc = scoring.AUC.pair(test_results, 1, 2) 
     87        auc = scoring.AUC_for_pair_of_classes(test_results, 0, 1) 
     88        self.assertEqual(len(auc), 1) 
     89        auc = scoring.AUC_for_pair_of_classes(test_results, 0, 2) 
     90        self.assertEqual(len(auc), 1) 
     91        auc = scoring.AUC_for_pair_of_classes(test_results, 1, 2) 
    9292        self.assertEqual(len(auc), 1) 
    9393 
     
    9595        ds = data.Table("iris") 
    9696        test_results = testing.cross_validation([self.learner], ds, folds=5) 
    97         auc = scoring.AUC.matrix(test_results) 
     97        auc = scoring.AUC_matrix(test_results) 
    9898        self.assertEqual(len(auc), 1) 
    9999        self.assertEqual(len(auc[0]), 3) 
     
    246246    def score(self): 
    247247        return scoring.MCC 
     248 
    248249if __name__ == '__main__': 
    249250    unittest.main() 
Note: See TracChangeset for help on using the changeset viewer.