161a91e9a65de139b1b3e54ff680b53a74610a48,tests/test_classifier/test_rocauc.py,ROCAUCTests,test_rocauc_no_macro_no_micro,#ROCAUCTests#,232

Before Change

        Test ROCAUC without a macro or micro average
        
        // Load the Data
        X_train, X_test, y_train, y_test = self.load_binary_data()

        // Create and fit the visualizer
        visualizer = ROCAUC(LogisticRegression(), macro=False, micro=False)
        visualizer.fit(X_train, y_train)

        // Score the visualizer (should be the F1 score)
        s = visualizer.score(X_test, y_test)
        self.assertAlmostEqual(s, 0.98978599221789887)

        // Assert that there is no macro score
        self.assertNotIn("macro", visualizer.fpr)

After Change

        
        // Create and fit the visualizer
        visualizer = ROCAUC(LogisticRegression(), macro=False, micro=False)
        visualizer.fit(self.binary.X.train, self.binary.y.train)

        // Score the visualizer (should be the F1 score)
        s = visualizer.score(self.binary.X.test, self.binary.y.test)
        self.assertAlmostEqual(s, 0.8)

        // Assert that there is no macro score
        self.assertNotIn("macro", visualizer.fpr)

Instances