def test_if_negative_evidence_is_distributed(self): stratifier = IterativeStratification(n_splits=2, order=1) y = np.matrix([[0, 0], [1, 0], [0, 1], [1, 1]]) rows, rows_used, all_combinations, per_row_combinations, samples_with_combination, folds = \ stratifier._prepare_stratification(y) stratifier._distribute_positive_evidence(rows_used, folds, samples_with_combination, per_row_combinations) self.assertFalse(rows_used[0]) stratifier._distribute_negative_evidence(rows_used, folds) self.assertTrue(rows_used[0])
def test_if_negative_evidence_is_distributed(self): stratifier = IterativeStratification(n_splits=2, order=1) y = np.matrix([[0, 0], [1, 0], [0, 1], [1, 1]]) rows, rows_used, all_combinations, per_row_combinations, samples_with_combination, folds = \ stratifier._prepare_stratification(y) stratifier._distribute_positive_evidence(rows_used, folds, samples_with_combination, per_row_combinations) self.assertFalse(rows_used[0]) stratifier._distribute_negative_evidence(rows_used, folds) self.assertTrue(rows_used[0])
def test_if_positive_evidence_does_not_include_negative_evidence(self): stratifier = IterativeStratification(n_splits=2, order=1) y = np.matrix([[0, 0], [1, 0], [0, 1], [1, 1]]) rows, rows_used, all_combinations, per_row_combinations, samples_with_combination, folds = \ stratifier._prepare_stratification(y) stratifier._distribute_positive_evidence(rows_used, folds, samples_with_combination, per_row_combinations) self.assertFalse(rows_used[0]) self.assertTrue(rows_used[1]) self.assertTrue(rows_used[2]) self.assertTrue(rows_used[3]) for combination, samples in stratifier.desired_samples_per_combination_per_fold.items(): for desire in samples: self.assertEqual(desire, 0)
def test_if_positive_evidence_does_not_include_negative_evidence(self): stratifier = IterativeStratification(n_splits=2, order=1) y = np.matrix([[0, 0], [1, 0], [0, 1], [1, 1]]) rows, rows_used, all_combinations, per_row_combinations, samples_with_combination, folds = \ stratifier._prepare_stratification(y) stratifier._distribute_positive_evidence(rows_used, folds, samples_with_combination, per_row_combinations) self.assertFalse(rows_used[0]) self.assertTrue(rows_used[1]) self.assertTrue(rows_used[2]) self.assertTrue(rows_used[3]) for combination, samples in stratifier.desired_samples_per_combination_per_fold.items( ): for desire in samples: self.assertEqual(desire, 0)