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

debugging

parent 286933af
Branches
Tags
No related merge requests found
...@@ -302,6 +302,7 @@ SVMLinearKWARGSInit = {"0":map(int, args.CL_SVML_C.split(":"))[0]} ...@@ -302,6 +302,7 @@ SVMLinearKWARGSInit = {"0":map(int, args.CL_SVML_C.split(":"))[0]}
SVMRBFKWARGSInit = {"0":map(int, args.CL_SVMR_C.split(":"))[0]} SVMRBFKWARGSInit = {"0":map(int, args.CL_SVMR_C.split(":"))[0]}
SVMPolyKWARGSInit = {"0":map(int, args.CL_SVMP_C.split(":"))[0], '1':map(int, args.CL_SVMP_deg.split(":"))[0]} SVMPolyKWARGSInit = {"0":map(int, args.CL_SVMP_C.split(":"))[0], '1':map(int, args.CL_SVMP_deg.split(":"))[0]}
DecisionTreeKWARGSInit = {"0":map(int, args.CL_DT_depth.split(":"))[0]} DecisionTreeKWARGSInit = {"0":map(int, args.CL_DT_depth.split(":"))[0]}
print DecisionTreeKWARGSInit
SGDKWARGSInit = {"2": map(float, args.CL_SGD_alpha.split(":"))[0], "1": args.CL_SGD_penalty.split(":")[0], SGDKWARGSInit = {"2": map(float, args.CL_SGD_alpha.split(":"))[0], "1": args.CL_SGD_penalty.split(":")[0],
"0":args.CL_SGD_loss.split(":")[0]} "0":args.CL_SGD_loss.split(":")[0]}
KNNKWARGSInit = {"0": map(float, args.CL_KNN_neigh.split(":"))[0]} KNNKWARGSInit = {"0": map(float, args.CL_KNN_neigh.split(":"))[0]}
......
...@@ -12,7 +12,6 @@ __status__ = "Prototype" # Production, Development, P ...@@ -12,7 +12,6 @@ __status__ = "Prototype" # Production, Development, P
def fit(DATASET, CLASS_LABELS, NB_CORES=1, **kwargs): def fit(DATASET, CLASS_LABELS, NB_CORES=1, **kwargs):
maxDepth = int(kwargs['0']) maxDepth = int(kwargs['0'])
print maxDepth
classifier = DecisionTreeClassifier(max_depth=maxDepth) classifier = DecisionTreeClassifier(max_depth=maxDepth)
classifier.fit(DATASET, CLASS_LABELS) classifier.fit(DATASET, CLASS_LABELS)
return classifier return classifier
......
...@@ -64,6 +64,7 @@ class Fusion: ...@@ -64,6 +64,7 @@ class Fusion:
fusionMethodClass = getattr(fusionMethodModule, fusionMethod) fusionMethodClass = getattr(fusionMethodModule, fusionMethod)
nbCores = NB_CORES nbCores = NB_CORES
classifierKWARGS = dict((key, value) for key, value in kwargs.iteritems() if key not in ['fusionType', 'fusionMethod']) classifierKWARGS = dict((key, value) for key, value in kwargs.iteritems() if key not in ['fusionType', 'fusionMethod'])
print classifierKWARGS
self.classifier = fusionMethodClass(NB_CORES=nbCores, **classifierKWARGS) self.classifier = fusionMethodClass(NB_CORES=nbCores, **classifierKWARGS)
def fit_hdf5(self, DATASET, trainIndices=None, viewsIndices=None): def fit_hdf5(self, DATASET, trainIndices=None, viewsIndices=None):
......
...@@ -9,6 +9,7 @@ class EarlyFusionClassifier(object): ...@@ -9,6 +9,7 @@ class EarlyFusionClassifier(object):
def __init__(self, monoviewClassifierName, monoviewClassifierConfig, NB_CORES=1): def __init__(self, monoviewClassifierName, monoviewClassifierConfig, NB_CORES=1):
self.monoviewClassifierName = monoviewClassifierName[0] self.monoviewClassifierName = monoviewClassifierName[0]
self.monoviewClassifiersConfig = monoviewClassifierConfig[0] self.monoviewClassifiersConfig = monoviewClassifierConfig[0]
print monoviewClassifierConfig
self.monoviewClassifier = None self.monoviewClassifier = None
self.nbCores = NB_CORES self.nbCores = NB_CORES
self.monoviewData = None self.monoviewData = None
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment