Skip to content
Snippets Groups Projects
Commit 9cc12c4b authored by bbauvin's avatar bbauvin
Browse files

Minor change

parent 1561fb4d
No related branches found
No related tags found
No related merge requests found
......@@ -62,7 +62,7 @@ def genParamsSets(classificationKWARGS, randomState, nIter=1):
max_attributes = randomState.randint(1, 20)
p = randomState.random_sample()
model = randomState.choice(["conjunction", "disjunction"])
order = randomState.randint(1, nbView)
order = randomState.randint(1, 10)
paramsSets.append([p, max_attributes, model, order])
return paramsSets
......@@ -123,7 +123,6 @@ class SCMForLinear(LateFusionClassifier):
self.p = paramsSet[0]
self.maxAttributes = paramsSet[1]
self.order = paramsSet[3]
self.order = 2
self.modelType = paramsSet[2]
def fit_hdf5(self, DATASET, trainIndices=None, viewsIndices=None):
......
......@@ -71,3 +71,13 @@ class Test_genSplits(unittest.TestCase):
self.assertGreater(len(np.where(self.labels[self.test_indices]==0)[0]), 0)
self.assertGreater(len(np.where(self.labels[self.train_indices]==1)[0]), 0)
self.assertGreater(len(np.where(self.labels[self.test_indices]==1)[0]), 0)
class Test_genKFolds(unittest.TestCase):
def setUp(self):
self.statsIter = 2
self.nbFolds = 5
self.statsIterRandomStates = [np.random.RandomState(42), np.random.RandomState(94)]
def test_genKFolds_iter(self):
pass
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment