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

debugging

parent f5c33df2
No related branches found
No related tags found
No related merge requests found
...@@ -11,6 +11,7 @@ __status__ = "Prototype" # Production, Development, P ...@@ -11,6 +11,7 @@ __status__ = "Prototype" # Production, Development, P
def fit(DATASET, CLASS_LABELS, NB_CORES=1, **kwargs): def fit(DATASET, CLASS_LABELS, NB_CORES=1, **kwargs):
print kwargs
maxDepth = int(kwargs['0']) maxDepth = int(kwargs['0'])
classifier = DecisionTreeClassifier(max_depth=maxDepth) classifier = DecisionTreeClassifier(max_depth=maxDepth)
classifier.fit(DATASET, CLASS_LABELS) classifier.fit(DATASET, CLASS_LABELS)
......
...@@ -37,6 +37,9 @@ class WeightedLinear(EarlyFusionClassifier): ...@@ -37,6 +37,9 @@ class WeightedLinear(EarlyFusionClassifier):
if not trainIndices: if not trainIndices:
trainIndices = range(DATASET.get("Metadata").attrs["datasetLength"]) trainIndices = range(DATASET.get("Metadata").attrs["datasetLength"])
self.weights = self.weights/float(max(self.weights)) self.weights = self.weights/float(max(self.weights))
print dict((str(configIndex), config) for configIndex, config in
enumerate(self.monoviewClassifiersConfig
))
self.makeMonoviewData_hdf5(DATASET, weights=self.weights, usedIndices=trainIndices, viewsIndices=viewsIndices) self.makeMonoviewData_hdf5(DATASET, weights=self.weights, usedIndices=trainIndices, viewsIndices=viewsIndices)
monoviewClassifierModule = getattr(MonoviewClassifiers, self.monoviewClassifierName) monoviewClassifierModule = getattr(MonoviewClassifiers, self.monoviewClassifierName)
self.monoviewClassifier = monoviewClassifierModule.fit(self.monoviewData, DATASET.get("Labels")[trainIndices], self.monoviewClassifier = monoviewClassifierModule.fit(self.monoviewData, DATASET.get("Labels")[trainIndices],
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment