Skip to content
Snippets Groups Projects
deeplearning 841 B
Newer Older
#!/usr/bin/python

from keras.models import Sequential
from keras.layers import Dense

import numpy
# Fix random seed for reproducibility
seed = 7
numpy.random.seed(seed)
Pizon Antoine's avatar
Pizon Antoine committed

# Loading data
Pizon Antoine's avatar
Pizon Antoine committed
dataset = numpy.loadtxt("save.csv", delimiter=";")
# Split into input (X) and output (Y) variables
Pizon Antoine's avatar
Pizon Antoine committed
X = dataset[:,0:5]
Pizon Antoine's avatar
Pizon Antoine committed
Y = dataset[:,5:]
Pizon Antoine's avatar
Pizon Antoine committed

# Creating the model
Pizon Antoine's avatar
Pizon Antoine committed
model = Sequential()
model.add(Dense(12, input_dim=5, init='uniform', activation='relu'))
model.add(Dense(8, init='uniform', activation='relu'))
Pizon Antoine's avatar
Pizon Antoine committed
model.add(Dense(2, init='uniform', activation='softmax'))
Pizon Antoine's avatar
Pizon Antoine committed

# Compiling the model
Pizon Antoine's avatar
Pizon Antoine committed
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
Pizon Antoine's avatar
Pizon Antoine committed

# Training the model
Pizon Antoine's avatar
Pizon Antoine committed
model.fit(X, Y, nb_epoch=150, batch_size=10)
Pizon Antoine's avatar
Pizon Antoine committed

# Evaluating the model
Pizon Antoine's avatar
Pizon Antoine committed
scores = model.evaluate(X, Y)
print("%s: %.2f%%" % (model.metrics_names[1], scores[1]*100))