- AlexNet
- All-CNN-C
- General DeepNeuralNetwork
- General 1D MultiChannel CNN
- General 2D MultiChannel CNN
- GoogLeNet (Inception v1)
- LeNet-5
- Variational AutoEncoder
- VGG16
- VGG19
- ZFNet
from keras.datasets import mnist
import sys
sys.append("Models/")
from AllCNN import AllCNN
(X_train, y_train), (X_test, y_test) = mnist.load_data()
model = AllCNN(X_train,y_train)
model.train(100)
predictions = model.predict(input)