Diff of /Serialized/defenitions.py [000000] .. [53d15f]

Switch to side-by-side view

--- a
+++ b/Serialized/defenitions.py
@@ -0,0 +1,113 @@
+data_dir = '/media/nvme/data/RSNA/'
+train_images_dir=data_dir+'stage_1_train_images/'
+test_images_dir=data_dir+'stage_1_test_images/'  #data_dir+'stage_2_test_images/'
+models_dir = '/media/hd/notebooks/data/RSNA/models/'
+outputs_dir = '/media/hd/notebooks/data/RSNA/outputs/'
+models_format='model_{}_version_{}_split_{}.pth'
+outputs_format='model_{}_version_{}_type_{}_split_{}.pkl'
+hemorrhage_types=['epidural','intraparenchymal','intraventricular','subarachnoid','subdural','any']
+bad_images = ['e20bf3f8a','8da38f2e4','6431af929','470e639ae','0e21abf7a','d91d52bdc']
+my_gmail='********'
+my_pass='*********'
+to_email='*********'
+parameters={ 'se_resnet101_3': {
+                            'model_name':'se_resnet101',
+                            'SEED':8153,
+                            'n_splits':3,
+                            'Pre_version':None,
+                            'focal':False,
+                            'version':'classifier_splits',
+                            'train_prediction':'predictions_train_tta',
+                            'train_features':'features_train_tta',
+                            'test_prediction':'predictions_test',
+                            'test_features':'features_test',
+                            'num_epochs' : 5,
+                            'num_pool':8
+                            },
+         'se_resnet101_5': {
+                            'model_name':'se_resnet101',
+                            'SEED':432,
+                            'n_splits':5,
+                            'Pre_version':None,
+                            'focal':False,
+                            'version':'new_splits',
+                            'train_prediction':'predictions_train_tta',
+                            'train_features':'features_train_tta',
+                            'test_prediction':'predictions_test',
+                            'test_features':'features_test',
+                            'num_epochs' : 5,
+                            'num_pool':8
+                            },
+         'se_resnet101_focal': {
+                            'model_name':'se_resnet101',
+                            'SEED':432,
+                            'n_splits':5,
+                            'Pre_version':'new_splits',
+                            'focal':True,
+                            'version':'new_splits_focal',
+                            'train_prediction':'predictions_train_tta',
+                            'train_features':'features_train_tta',
+                            'test_prediction':'predictions_test',
+                            'test_features':'features_test',
+                            'num_epochs' : 5,
+                            'num_pool':8
+                            },
+         'se_resnext101_32x4d_3': {
+                            'model_name':'se_resnext101_32x4d',
+                            'SEED':8153,
+                            'n_splits':3,
+                            'Pre_version':None,
+                            'focal':False,
+                            'version':'classifier_splits',
+                            'train_prediction':'predictions_train_tta',
+                            'train_features':'features_train_tta',
+                            'test_prediction':'predictions_test_tta',
+                            'test_features':'features_test_tta',
+                            'num_epochs' : 5,
+                            'num_pool':8
+                            },
+         'se_resnext101_32x4d_5': {
+                            'model_name':'se_resnext101_32x4d',
+                            'SEED':432,
+                            'n_splits':5,
+                            'Pre_version':None,
+                            'focal':False,
+                            'version':'new_splits',
+                            'train_prediction':'predictions_train_tta',
+                            'train_features':'features_train_tta',
+                            'test_prediction':'predictions_test',
+                            'test_features':'features_test',
+                            'num_epochs' : 5,
+                            'num_pool':8
+                            },
+         'Densenet161_3': {
+                            'model_name':'Densenet161_3',
+                            'SEED':8153,
+                            'n_splits':3,
+                            'Pre_version':None,
+                            'focal':False,
+                            'version':'classifier_splits',
+                            'train_prediction':'predictions_train_tta2',
+                            'train_features':'features_train_tta2',
+                            'test_prediction':'predictions_test',
+                            'test_features':'features_test',
+                            'num_epochs' : 5,
+                            'num_pool':4
+                            },
+         'Densenet169_3': {
+                            'model_name':'Densenet169_3',
+                            'SEED':8153,
+                            'n_splits':3,
+                            'Pre_version':None,
+                            'focal':False,
+                            'version':'classifier_splits',
+                            'train_prediction':'predictions_train_tta2',
+                            'train_features':'features_train_tta2',
+                            'test_prediction':'predictions_test',
+                            'test_features':'features_test',
+                            'num_epochs' : 5,
+                            'num_pool':8
+                            }
+        
+       }
+