20 lines
377 B
Python
20 lines
377 B
Python
import tensorflow as tf
|
|
import tensorflow.keras.layers as l
|
|
from sys import argv
|
|
|
|
import f
|
|
|
|
m = tf.keras.models.Sequential([
|
|
l.Input(shape = (784,)),
|
|
l.Dense(64, activation = "relu"),
|
|
l.Dense(20, activation = "relu"),
|
|
l.Dense(10, activation = "softmax")
|
|
])
|
|
|
|
#f.train(m, "2")
|
|
|
|
if len(argv) == 2:
|
|
f.classify(m, "2", argv[1])
|
|
else:
|
|
f.classify_live(m, "2")
|