/
model_playground.py
29 lines (25 loc) · 1.09 KB
/
model_playground.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
from feat_extract import saveMFCCForWavs, AnnotatedSongLabeler, generateLabeledFeatures, Pooling1DFeature, saveFeatures
from feat_extract import generateTrainingDataForAudio
from feat_extract import generateTrainingDataForAudios
from ml import stupidSimpleRNNModel, modelV4
from ml import stupidSimpleRNNModelTimeDisLess
from ml import dumbSimpleRNNModel
from ml import trainWithModelSingleSong
from ml import trainModel
from ml import trainModelWithGenerator
from ml import modelV1
from feat_extract import getFeatsAndClassificationsFromFile
from feat_extract import trainingGeneratorFromFolder
from keras.models import load_model
from numpy import zeros, newaxis
from feat_extract import MFCCFeature
numSegmentTypes = 6
load = False
modelname = "pooled_model_v4.h5"
if load:
model = load_model(modelname)
else:
model = modelV4(inputDimension=200, numPerRecurrentLayer=75, numRecurrentLayers=2,
outputDimension=numSegmentTypes, kernelSize=10, stride=1)
model.summary()
trainModelWithGenerator(model, trainingGeneratorFromFolder, "features", modelname, 30)