new experiment

urppchusj · Jul 27, 2020 · f4d77b6 · f4d77b6
1 parent 3a20075
commit f4d77b6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/evaluate_outliers.py b/evaluate_outliers.py
@@ -15,7 +15,7 @@
 n_folds = 3
 depas_to_score = ['Overall', 'Néonatologie', 'Ob/gyn', 'Oncologie', 'Pédiatrie']
 
-anomaly_algorithm_rename_dict = {'LocalOutlierFactor(contamination={}, novelty=True)'.format(contamination_ratio):'LOF', 'IsolationForest(contamination={})'.format(contamination_ratio):'IF', 'OneClassSVM(nu={})'.format(contamination_ratio):'OC SVM gamma scale', 'OneClassSVM(gamma=\'auto\', nu={})'.format(contamination_ratio):'OC SVM gamma auto'}
+anomaly_algorithm_rename_dict = {'LocalOutlierFactor(contamination={}, novelty=True)'.format(contamination_ratio):'LOF', 'IsolationForest(contamination={})'.format(contamination_ratio):'IF', 'OneClassSVM(nu={})'.format(contamination_ratio):'OC SVM', 'EllipticEnvelope(contamination={}'.format(contamination_ratio):'EE'}
 
 x_axis_name_in_figure = 'Components'
 algorith_name_in_figure = 'Alg'
@@ -55,6 +55,7 @@ def make_variance_graphs(df):
 	sns.set(style="whitegrid", font_scale=2)
 	f = sns.catplot(font_scale = 2, x=x_axis_name_in_figure, y="Explained variance ratio", data=graph_df, kind='point')
 	f.set_xticklabels(rotation=35,  horizontalalignment='right')
+	plt.ylim(0,1)
 	plt.savefig(os.path.join(data_dir, 'variance_ratio_results.png'))
 
 

diff --git a/train_outliers.py b/train_outliers.py
@@ -42,7 +42,7 @@
 contamination_ratio = 0.2
 param_grid = dict(
 				tsvd__n_components = [8,16,32,124,256,512,1024],
-				anomaly_algorithm = [IsolationForest(contamination=contamination_ratio)]#, OneClassSVM(nu=contamination_ratio, gamma='scale')], #EllipticEnvelope(contamination=contamination_ratio), OneClassSVM(nu=contamination_ratio, gamma='auto'), LocalOutlierFactor(novelty=True, contamination=contamination_ratio), 
+				anomaly_algorithm = [IsolationForest(contamination=contamination_ratio), OneClassSVM(nu=contamination_ratio)], #EllipticEnvelope(contamination=contamination_ratio) , LocalOutlierFactor(novelty=True, contamination=contamination_ratio), 
 				)
 tsvd_n_components = 256