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

debugging

parent 286933af
No related branches found
No related tags found
No related merge requests found
......@@ -302,6 +302,7 @@ SVMLinearKWARGSInit = {"0":map(int, args.CL_SVML_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]}
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],
"0":args.CL_SGD_loss.split(":")[0]}
KNNKWARGSInit = {"0": map(float, args.CL_KNN_neigh.split(":"))[0]}
......
......@@ -12,7 +12,6 @@ __status__ = "Prototype" # Production, Development, P
def fit(DATASET, CLASS_LABELS, NB_CORES=1, **kwargs):
maxDepth = int(kwargs['0'])
print maxDepth
classifier = DecisionTreeClassifier(max_depth=maxDepth)
classifier.fit(DATASET, CLASS_LABELS)
return classifier
......
......@@ -64,6 +64,7 @@ class Fusion:
fusionMethodClass = getattr(fusionMethodModule, fusionMethod)
nbCores = NB_CORES
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)
def fit_hdf5(self, DATASET, trainIndices=None, viewsIndices=None):
......
......@@ -9,6 +9,7 @@ class EarlyFusionClassifier(object):
def __init__(self, monoviewClassifierName, monoviewClassifierConfig, NB_CORES=1):
self.monoviewClassifierName = monoviewClassifierName[0]
self.monoviewClassifiersConfig = monoviewClassifierConfig[0]
print monoviewClassifierConfig
self.monoviewClassifier = None
self.nbCores = NB_CORES
self.monoviewData = None
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment