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

debugging

parent f5c33df2
Branches
Tags
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