def test_if_order_is_set(self):
classifier = ClassifierChain(
classifier=GaussianNB(), require_dense=[True, True], order=None
)
X, y = self.get_multilabel_data_for_tests(sparsity_indicator='sparse')[0]
classifier.fit(X,y)
self.assertEqual(classifier._order(), list(range(y.shape[1])))
def test_if_order_is_set(self):
classifier = ClassifierChain(classifier=GaussianNB(),
require_dense=[True, True],
order=None)
X, y = self.get_multilabel_data_for_tests(
sparsity_indicator='sparse')[0]
classifier.fit(X, y)
self.assertEqual(classifier._order(), list(range(y.shape[1])))
def test_if_order_is_set_when_explicitly_given(self):
X, y = self.get_multilabel_data_for_tests(sparsity_indicator='sparse')[0]
reversed_chain = list(reversed(range(y.shape[1])))
classifier = ClassifierChain(
classifier=GaussianNB(), require_dense=[True, True], order=reversed_chain
)
classifier.fit(X, y)
self.assertEqual(classifier._order(), reversed_chain)
def test_if_order_is_set_when_explicitly_given(self):
X, y = self.get_multilabel_data_for_tests(
sparsity_indicator='sparse')[0]
reversed_chain = list(reversed(range(y.shape[1])))
classifier = ClassifierChain(classifier=GaussianNB(),
require_dense=[True, True],
order=reversed_chain)
classifier.fit(X, y)
self.assertEqual(classifier._order(), reversed_chain)
