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

debugging

parent 286933af
Branches
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