-
Notifications
You must be signed in to change notification settings - Fork 0
/
constants.py
107 lines (99 loc) · 8.56 KB
/
constants.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
UNIVARIATE_DATASET_NAMES = ['50words', 'Adiac', 'ArrowHead', 'Beef', 'BeetleFly', 'BirdChicken', 'Car', 'CBF',
'ChlorineConcentration', 'CinC_ECG_torso', 'Coffee',
'Computers', 'Cricket_X', 'Cricket_Y', 'Cricket_Z', 'DiatomSizeReduction',
'DistalPhalanxOutlineAgeGroup', 'DistalPhalanxOutlineCorrect', 'DistalPhalanxTW',
'Earthquakes', 'ECG200', 'ECG5000', 'ECGFiveDays', 'ElectricDevices', 'FaceAll', 'FaceFour',
'FacesUCR', 'FISH', 'FordA', 'FordB', 'Gun_Point', 'Ham', 'HandOutlines',
'Haptics', 'Herring', 'InlineSkate', 'InsectWingbeatSound', 'ItalyPowerDemand',
'LargeKitchenAppliances', 'Lighting2', 'Lighting7', 'MALLAT', 'Meat', 'MedicalImages',
'MiddlePhalanxOutlineAgeGroup', 'MiddlePhalanxOutlineCorrect', 'MiddlePhalanxTW',
'MoteStrain', 'NonInvasiveFatalECG_Thorax1', 'NonInvasiveFatalECG_Thorax2', 'OliveOil',
'OSULeaf', 'PhalangesOutlinesCorrect', 'Phoneme', 'Plane', 'ProximalPhalanxOutlineAgeGroup',
'ProximalPhalanxOutlineCorrect', 'ProximalPhalanxTW', 'RefrigerationDevices',
'ScreenType', 'ShapeletSim', 'ShapesAll', 'SmallKitchenAppliances', 'SonyAIBORobotSurface',
'SonyAIBORobotSurfaceII', 'StarLightCurves', 'Strawberry', 'SwedishLeaf', 'Symbols',
'synthetic_control', 'ToeSegmentation1', 'ToeSegmentation2', 'Trace', 'TwoLeadECG',
'Two_Patterns', 'UWaveGestureLibraryAll', 'uWaveGestureLibrary_X', 'uWaveGestureLibrary_Y',
'uWaveGestureLibrary_Z', 'wafer', 'Wine', 'WordsSynonyms', 'Worms', 'WormsTwoClass', 'yoga']
UNIVARIATE_DATASET_NAMES_2018 = ['ACSF1', 'Adiac', 'AllGestureWiimoteX', 'AllGestureWiimoteY', 'AllGestureWiimoteZ',
'ArrowHead', 'Beef', 'BeetleFly', 'BirdChicken', 'BME', 'Car', 'CBF', 'Chinatown',
'ChlorineConcentration', 'CinCECGTorso', 'Coffee', 'Computers', 'CricketX',
'CricketY', 'CricketZ', 'Crop', 'DiatomSizeReduction',
'DistalPhalanxOutlineAgeGroup', 'DistalPhalanxOutlineCorrect', 'DistalPhalanxTW',
'DodgerLoopDay', 'DodgerLoopGame', 'DodgerLoopWeekend', 'Earthquakes', 'ECG200',
'ECG5000', 'ECGFiveDays', 'ElectricDevices', 'EOGHorizontalSignal',
'EOGVerticalSignal', 'EthanolLevel', 'FaceAll', 'FaceFour', 'FacesUCR',
'FiftyWords', 'Fish', 'FordA', 'FordB', 'FreezerRegularTrain',
'FreezerSmallTrain', 'Fungi', 'GestureMidAirD1', 'GestureMidAirD2',
'GestureMidAirD3', 'GesturePebbleZ1', 'GesturePebbleZ2', 'GunPoint',
'GunPointAgeSpan', 'GunPointMaleVersusFemale', 'GunPointOldVersusYoung',
'Ham', 'HandOutlines', 'Haptics', 'Herring', 'HouseTwenty', 'InlineSkate',
'InsectEPGRegularTrain', 'InsectEPGSmallTrain', 'InsectWingbeatSound',
'ItalyPowerDemand', 'LargeKitchenAppliances', 'Lightning2', 'Lightning7',
'Mallat', 'Meat', 'MedicalImages', 'MelbournePedestrian',
'MiddlePhalanxOutlineAgeGroup', 'MiddlePhalanxOutlineCorrect',
'MiddlePhalanxTW', 'MixedShapesRegularTrain', 'MixedShapesSmallTrain',
'MoteStrain', 'NonInvasiveFetalECGThorax1', 'NonInvasiveFetalECGThorax2',
'OliveOil', 'OSULeaf', 'PhalangesOutlinesCorrect', 'Phoneme',
'PickupGestureWiimoteZ', 'PigAirwayPressure', 'PigArtPressure', 'PigCVP',
'PLAID', 'Plane', 'PowerCons', 'ProximalPhalanxOutlineAgeGroup',
'ProximalPhalanxOutlineCorrect', 'ProximalPhalanxTW', 'RefrigerationDevices',
'Rock', 'ScreenType', 'SemgHandGenderCh2', 'SemgHandMovementCh2',
'SemgHandSubjectCh2', 'ShakeGestureWiimoteZ', 'ShapeletSim', 'ShapesAll',
'SmallKitchenAppliances', 'SmoothSubspace', 'SonyAIBORobotSurface1',
'SonyAIBORobotSurface2', 'StarLightCurves', 'Strawberry', 'SwedishLeaf',
'Symbols', 'SyntheticControl', 'ToeSegmentation1', 'ToeSegmentation2', 'Trace',
'TwoLeadECG', 'TwoPatterns', 'UMD', 'UWaveGestureLibraryAll',
'UWaveGestureLibraryX', 'UWaveGestureLibraryY', 'UWaveGestureLibraryZ',
'Wafer', 'Wine', 'WordSynonyms', 'Worms', 'WormsTwoClass', 'Yoga']
MTS_DATASET_NAMES = ['ArabicDigits', 'AUSLAN', 'CharacterTrajectories', 'CMUsubject16', 'ECG',
'JapaneseVowels', 'KickvsPunch', 'Libras', 'NetFlow', 'UWave', 'Wafer', 'WalkvsRun']
ITERATIONS = 5 # nb of random runs for random initializations
ARCHIVE_NAMES = ['UCRArchive_2018']
dataset_names_for_archive = {'UCRArchive_2018': UNIVARIATE_DATASET_NAMES_2018}
CLASSIFIERS = ['fcn', 'mlp', 'resnet', 'tlenet', 'mcnn', 'twiesn', 'encoder', 'mcdcnn', 'cnn', 'inception']
dataset_types = {'ElectricDevices': 'DEVICE', 'FordB': 'SENSOR',
'FordA': 'SENSOR', 'NonInvasiveFatalECG_Thorax2': 'ECG',
'NonInvasiveFatalECG_Thorax1': 'ECG', 'PhalangesOutlinesCorrect': 'IMAGE',
'HandOutlines': 'IMAGE', 'StarLightCurves': 'SENSOR',
'wafer': 'SENSOR', 'Two_Patterns': 'SIMULATED',
'UWaveGestureLibraryAll': 'MOTION', 'uWaveGestureLibrary_Z': 'MOTION',
'uWaveGestureLibrary_Y': 'MOTION', 'uWaveGestureLibrary_X': 'MOTION',
'Strawberry': 'SPECTRO', 'ShapesAll': 'IMAGE',
'ProximalPhalanxOutlineCorrect': 'IMAGE', 'MiddlePhalanxOutlineCorrect': 'IMAGE',
'DistalPhalanxOutlineCorrect': 'IMAGE', 'FaceAll': 'IMAGE',
'ECG5000': 'ECG', 'SwedishLeaf': 'IMAGE', 'ChlorineConcentration': 'SIMULATED',
'50words': 'IMAGE', 'ProximalPhalanxTW': 'IMAGE', 'ProximalPhalanxOutlineAgeGroup': 'IMAGE',
'MiddlePhalanxOutlineAgeGroup': 'IMAGE', 'DistalPhalanxTW': 'IMAGE',
'DistalPhalanxOutlineAgeGroup': 'IMAGE', 'MiddlePhalanxTW': 'IMAGE',
'Cricket_Z': 'MOTION', 'Cricket_Y': 'MOTION',
'Cricket_X': 'MOTION', 'Adiac': 'IMAGE',
'MedicalImages': 'IMAGE', 'SmallKitchenAppliances': 'DEVICE',
'ScreenType': 'DEVICE', 'RefrigerationDevices': 'DEVICE',
'LargeKitchenAppliances': 'DEVICE', 'Earthquakes': 'SENSOR',
'yoga': 'IMAGE', 'synthetic_control': 'SIMULATED',
'WordsSynonyms': 'IMAGE', 'Computers': 'DEVICE',
'InsectWingbeatSound': 'SENSOR', 'Phoneme': 'SENSOR',
'OSULeaf': 'IMAGE', 'FacesUCR': 'IMAGE',
'WormsTwoClass': 'MOTION', 'Worms': 'MOTION',
'FISH': 'IMAGE', 'Haptics': 'MOTION',
'Epilepsy': 'HAR', 'Ham': 'SPECTRO',
'Plane': 'SENSOR', 'InlineSkate': 'MOTION',
'Trace': 'SENSOR', 'ECG200': 'ECG',
'Lighting7': 'SENSOR', 'ItalyPowerDemand': 'SENSOR',
'Herring': 'IMAGE', 'Lighting2': 'SENSOR',
'Car': 'SENSOR', 'Meat': 'SPECTRO',
'Wine': 'SPECTRO', 'MALLAT': 'SIMULATED',
'Gun_Point': 'MOTION', 'CinC_ECG_torso': 'ECG',
'ToeSegmentation1': 'MOTION', 'ToeSegmentation2': 'MOTION',
'ArrowHead': 'IMAGE', 'OliveOil': 'SPECTRO',
'Beef': 'SPECTRO', 'CBF': 'SIMULATED',
'Coffee': 'SPECTRO', 'SonyAIBORobotSurfaceII': 'SENSOR',
'Symbols': 'IMAGE', 'FaceFour': 'IMAGE',
'ECGFiveDays': 'ECG', 'TwoLeadECG': 'ECG',
'BirdChicken': 'IMAGE', 'BeetleFly': 'IMAGE',
'ShapeletSim': 'SIMULATED', 'MoteStrain': 'SENSOR',
'SonyAIBORobotSurface': 'SENSOR', 'DiatomSizeReduction': 'IMAGE'}
themes_colors = {'IMAGE': 'red', 'SENSOR': 'blue', 'ECG': 'green',
'SIMULATED': 'yellow', 'SPECTRO': 'orange',
'MOTION': 'purple', 'DEVICE': 'gray'}