elifmodel_type =='lstm': net = tflearn.lstm(net,128,dropout=0.8) #CNN elifmodel_type =='cnn': net = conv_model(net) net = tflearn.fully_connected(net,3,activation='softmax') net = tflearn.regression(net,optimizer='adam',learning_rate=0.001, loss='categorical_crossentropy') model = ...
elifmodel_type =='lstm': net = tflearn.lstm(net,128,dropout=0.8) #CNN elifmodel_type =='cnn': net = conv_model(net) net = tflearn.fully_connected(net,3,activation='softmax') net = tflearn.regression(net,optimizer='adam',learning_rate=0.001, loss='categorical_crossentropy') model = ...