-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathNNClassifier.py
More file actions
45 lines (38 loc) · 1.52 KB
/
NNClassifier.py
File metadata and controls
45 lines (38 loc) · 1.52 KB
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
from keras.models import Sequential
from keras.layers import Dense, Activation
from keras.utils import np_utils
import numpy as np
from sklearn.model_selection import train_test_split
import scipy.io
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.metrics import accuracy_score
from sklearn.preprocessing import LabelEncoder
from sklearn.model_selection import cross_val_score, KFold
from keras.wrappers.scikit_learn import KerasClassifier
data = pd.read_csv('./data/TrainData_Labeled.csv')
# Convert data from Pandas DataFrame to np array for X and y
X = np.asarray(data)[:, :11]
y = np.asarray(data)[:, 11]
encoder = LabelEncoder()
encoder.fit(y)
encoded_y = encoder.transform(y)
# convert integers to dummy variables (i.e. one hot encoded)
dummy_y = np_utils.to_categorical(encoded_y)
# Build Keras model
n_cols = X.shape[1]
def make_model():
model = Sequential()
# Input layer
model.add(Dense(11, activation='exponential', input_shape=(n_cols,)))
# Hidden layers
model.add(Dense(64, activation='relu'))
model.add(Dense(32, activation='relu'))
model.add(Dense(16, activation='relu'))
model.add(Dense(6, activation='softmax'))
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
return model
estimator = KerasClassifier(build_fn=make_model, epochs=20, batch_size=5, verbose=0)
kfold = KFold(n_splits=4, shuffle=True)
results = cross_val_score(estimator, X, dummy_y, cv=kfold)
print("Baseline: %.2f%% (%.2f%%)" % (results.mean()*100, results.std()*100))