Skip to content

Commit 6a217f0

Browse files
committed
-Improved dataset random split
-Early stop if accuracy get above certain value (train) -I didnt add the full dataset to the git to keep it lightweight (i will be adding download links)
1 parent effdddf commit 6a217f0

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

runExp.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -9,15 +9,15 @@
99
os.makedirs(outReport)
1010

1111

12-
datasets = ["MFPT32","MFPT96","CWRbearings",'MFPTFFT32','MFPTFFT96']
12+
# datasets = ["MFPT32","MFPT96","CWRbearings",'MFPTFFT32','MFPTFFT96']
1313

14-
# dataset =["CW32Scalograms"
15-
# "CW32Spectrograms",
16-
# "CW96Scalograms",
17-
# "MFPT32Scalograms",
18-
# "MFPT32Spectrograms",
19-
# "MFPT96Scalograms",
20-
# "MFPT96Spectrogram"]
14+
datasets =["CW32Scalograms"
15+
"CW32Spectrograms",
16+
"CW96Scalograms",
17+
"MFPT32Scalograms",
18+
"MFPT32Spectrograms",
19+
"MFPT96Scalograms",
20+
"MFPT96Spectrogram"]
2121

2222

2323

@@ -28,7 +28,7 @@
2828
alt = int(i)
2929

3030
now = time.strftime('-day%Y-%m-%d-time%H-%M')
31-
nb_kwargs = {'dataFolder': dFolder, 'alternativeArc': alt, 'timeNow': now, 'epochs' : 1}
31+
nb_kwargs = {'dataFolder': dFolder, 'alternativeArc': alt, 'timeNow': now, 'epochs' : 20}
3232

3333
dname = nb_kwargs['dataFolder'].split("/")[1]
3434
alt2 = str(nb_kwargs['alternativeArc'])

0 commit comments

Comments
 (0)