There are two models, with linear and CNN, respectively. VAE is with encoder/ decoder.
classic MNIST handwritten digit dataset, which contains digit 0-9 images of size 28*28. We build a logistic regression model to predict the class of digit.
Python
- Yichen Zhang (Professor)
- Li-Ci Chuang
- matplotlib.pyplot
- pandas
- numpy
- tensorflow.keras
- tensorflow.keras.layers
- tensorflow.keras.optimizers
- tensorflow.keras.models
- mse, binary_crossentropy from tensorflow.keras.losses
- tensorflow.keras.backend
- plot_model from tensorflow.keras.utils
- time
- mnist from tensorflow.keras.datasets #open dataset