diff --git a/Code/MonoMultiViewClassifiers/MonoviewClassifiers/DecisionTree.py b/Code/MonoMultiViewClassifiers/MonoviewClassifiers/DecisionTree.py index 1dbd83adeaef9bbd9afd130a6914e1c5de4e337c..5fe7d7488c9085eaeaa4136d84fa9b57218cfb41 100644 --- a/Code/MonoMultiViewClassifiers/MonoviewClassifiers/DecisionTree.py +++ b/Code/MonoMultiViewClassifiers/MonoviewClassifiers/DecisionTree.py @@ -3,7 +3,7 @@ from sklearn.pipeline import Pipeline # Pipelining in classification from sklearn.model_selection import RandomizedSearchCV from scipy.stats import randint import numpy as np -import graphviz +# import graphviz # import cPickle from .. import Metrics @@ -91,7 +91,7 @@ def getConfig(config): def getInterpret(classifier, directory): dot_data = tree.export_graphviz(classifier, out_file=None) - graph = graphviz.Source(dot_data) - graph.render(directory+"-tree.pdf") + # graph = graphviz.Source(dot_data) + # graph.render(directory+"-tree.pdf") interpretString = getFeatureImportance(classifier, directory) return interpretString \ No newline at end of file diff --git a/Code/Versions.py b/Code/Versions.py index b3a02c11d414b09a770d7ad7c7460fef25a54ea9..bd0becc0f5148821458ab5a428df7d3b568b74e1 100644 --- a/Code/Versions.py +++ b/Code/Versions.py @@ -96,11 +96,11 @@ def testVersions(): isUpToDate = False toInstall.append("h5py") - try: - import graphviz # - except: - isUpToDate = False - toInstall.append("graphviz") + # try: + # import graphviz # + # except: + # isUpToDate = False + # toInstall.append("graphviz") if not isUpToDate: print "You can't run at the moment, please install the following modules : \n"+ "\n".join(toInstall)