Datasets
Models
Downloads: 1
Diff of
/Serialized/defenitions.py
[000000]
..
[53d15f]
Switch to unified view
| a | b/Serialized/defenitions.py | ||
|---|---|---|---|
| 1 | data_dir = '/media/nvme/data/RSNA/' |
||
| 2 | train_images_dir=data_dir+'stage_1_train_images/' |
||
| 3 | test_images_dir=data_dir+'stage_1_test_images/' #data_dir+'stage_2_test_images/' |
||
| 4 | models_dir = '/media/hd/notebooks/data/RSNA/models/' |
||
| 5 | outputs_dir = '/media/hd/notebooks/data/RSNA/outputs/' |
||
| 6 | models_format='model_{}_version_{}_split_{}.pth' |
||
| 7 | outputs_format='model_{}_version_{}_type_{}_split_{}.pkl' |
||
| 8 | hemorrhage_types=['epidural','intraparenchymal','intraventricular','subarachnoid','subdural','any'] |
||
| 9 | bad_images = ['e20bf3f8a','8da38f2e4','6431af929','470e639ae','0e21abf7a','d91d52bdc'] |
||
| 10 | my_gmail='********' |
||
| 11 | my_pass='*********' |
||
| 12 | to_email='*********' |
||
| 13 | parameters={ 'se_resnet101_3': { |
||
| 14 | 'model_name':'se_resnet101', |
||
| 15 | 'SEED':8153, |
||
| 16 | 'n_splits':3, |
||
| 17 | 'Pre_version':None, |
||
| 18 | 'focal':False, |
||
| 19 | 'version':'classifier_splits', |
||
| 20 | 'train_prediction':'predictions_train_tta', |
||
| 21 | 'train_features':'features_train_tta', |
||
| 22 | 'test_prediction':'predictions_test', |
||
| 23 | 'test_features':'features_test', |
||
| 24 | 'num_epochs' : 5, |
||
| 25 | 'num_pool':8 |
||
| 26 | }, |
||
| 27 | 'se_resnet101_5': { |
||
| 28 | 'model_name':'se_resnet101', |
||
| 29 | 'SEED':432, |
||
| 30 | 'n_splits':5, |
||
| 31 | 'Pre_version':None, |
||
| 32 | 'focal':False, |
||
| 33 | 'version':'new_splits', |
||
| 34 | 'train_prediction':'predictions_train_tta', |
||
| 35 | 'train_features':'features_train_tta', |
||
| 36 | 'test_prediction':'predictions_test', |
||
| 37 | 'test_features':'features_test', |
||
| 38 | 'num_epochs' : 5, |
||
| 39 | 'num_pool':8 |
||
| 40 | }, |
||
| 41 | 'se_resnet101_focal': { |
||
| 42 | 'model_name':'se_resnet101', |
||
| 43 | 'SEED':432, |
||
| 44 | 'n_splits':5, |
||
| 45 | 'Pre_version':'new_splits', |
||
| 46 | 'focal':True, |
||
| 47 | 'version':'new_splits_focal', |
||
| 48 | 'train_prediction':'predictions_train_tta', |
||
| 49 | 'train_features':'features_train_tta', |
||
| 50 | 'test_prediction':'predictions_test', |
||
| 51 | 'test_features':'features_test', |
||
| 52 | 'num_epochs' : 5, |
||
| 53 | 'num_pool':8 |
||
| 54 | }, |
||
| 55 | 'se_resnext101_32x4d_3': { |
||
| 56 | 'model_name':'se_resnext101_32x4d', |
||
| 57 | 'SEED':8153, |
||
| 58 | 'n_splits':3, |
||
| 59 | 'Pre_version':None, |
||
| 60 | 'focal':False, |
||
| 61 | 'version':'classifier_splits', |
||
| 62 | 'train_prediction':'predictions_train_tta', |
||
| 63 | 'train_features':'features_train_tta', |
||
| 64 | 'test_prediction':'predictions_test_tta', |
||
| 65 | 'test_features':'features_test_tta', |
||
| 66 | 'num_epochs' : 5, |
||
| 67 | 'num_pool':8 |
||
| 68 | }, |
||
| 69 | 'se_resnext101_32x4d_5': { |
||
| 70 | 'model_name':'se_resnext101_32x4d', |
||
| 71 | 'SEED':432, |
||
| 72 | 'n_splits':5, |
||
| 73 | 'Pre_version':None, |
||
| 74 | 'focal':False, |
||
| 75 | 'version':'new_splits', |
||
| 76 | 'train_prediction':'predictions_train_tta', |
||
| 77 | 'train_features':'features_train_tta', |
||
| 78 | 'test_prediction':'predictions_test', |
||
| 79 | 'test_features':'features_test', |
||
| 80 | 'num_epochs' : 5, |
||
| 81 | 'num_pool':8 |
||
| 82 | }, |
||
| 83 | 'Densenet161_3': { |
||
| 84 | 'model_name':'Densenet161_3', |
||
| 85 | 'SEED':8153, |
||
| 86 | 'n_splits':3, |
||
| 87 | 'Pre_version':None, |
||
| 88 | 'focal':False, |
||
| 89 | 'version':'classifier_splits', |
||
| 90 | 'train_prediction':'predictions_train_tta2', |
||
| 91 | 'train_features':'features_train_tta2', |
||
| 92 | 'test_prediction':'predictions_test', |
||
| 93 | 'test_features':'features_test', |
||
| 94 | 'num_epochs' : 5, |
||
| 95 | 'num_pool':4 |
||
| 96 | }, |
||
| 97 | 'Densenet169_3': { |
||
| 98 | 'model_name':'Densenet169_3', |
||
| 99 | 'SEED':8153, |
||
| 100 | 'n_splits':3, |
||
| 101 | 'Pre_version':None, |
||
| 102 | 'focal':False, |
||
| 103 | 'version':'classifier_splits', |
||
| 104 | 'train_prediction':'predictions_train_tta2', |
||
| 105 | 'train_features':'features_train_tta2', |
||
| 106 | 'test_prediction':'predictions_test', |
||
| 107 | 'test_features':'features_test', |
||
| 108 | 'num_epochs' : 5, |
||
| 109 | 'num_pool':8 |
||
| 110 | } |
||
| 111 | |||
| 112 | } |
||
| 113 |