-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathletterTest.py
More file actions
36 lines (30 loc) · 953 Bytes
/
letterTest.py
File metadata and controls
36 lines (30 loc) · 953 Bytes
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
import pandas as pd
import tensorflow as tf
import numpy as np
from tensorflow import keras
from tensorflow.keras import layers
from tensorflow.keras.models import Model
dataRaw=pd.read_csv("letter-recognition.csv")
dataRaw.head()
dataRaw.tail()
print("Dimensions: ",dataRaw.shape,"\n")
print(dataRaw.info())
dataRaw.head()
let=list(np.sort(dataRaw['letter'].unique()))
print(let)
print(dataRaw.columns)
(xTrain,yTrain),(xTest,yTest)=mnist.load_data()
xTrain=xTrain/255
xTest=xTest/255
model=tf.keras.models.Sequential([
tf.keras.layers.Flatten(input_shape=(28,28)),
tf.keras.layers.Dense(128,activation='relu'),
tf.keras.layers.Dropout(.2),
tf.keras.layers.Dense(32),
tf.keras.layers.Dropout(.2),
tf.keras.layers.Dense(10,activation='softmax')
])
model.compile(optimizer='adam',loss='sparse_categorical_crossentropy',metrics=['accuracy'])
model.fit(xTrain,yTrain,epochs=10)
model.evaluate(xTest,yTest)
model.summary()