Skip to content

This is a small project which aims to show an example of applied machine learning in Python 3 with the Keras library and its TensorFlow backend to train a neural network model for it to learn to add two integers.

License

PhTrempe/l2a

Repository files navigation

l2a (Learn to Add)

Description

This is a small project which aims to show an example of applied machine learning in Python 3 with the Keras library and its TensorFlow backend to train a neural network model for it to learn to add two integers.

The project also aims to follow the 7 Steps of Machine Learning presented by Google in this YouTube video.

  1. Gathering Data (dataset_generator.py)
  2. Preparing Data (dataset_preparer.py)
  3. Choosing a Model (model_builder.py)
  4. Training (trainer.py)
  5. Evaluation (trainer.py)
  6. Hyperparameter Tuning (hyperparameters.py)
  7. Prediction (predictor.py)

Usage

Installing Dependencies

conda install numpy scipy
pip install tensorflow tensorflow-gpu keras h5py

Running the Training Process

python trainer.py

This will first generate a dataset if none exists yet. It will then prepare the dataset if no prepared dataset exists yet. After that, it will build the model using the model builder (cf. model_builder.py) if no model exists yet. If an existing model is found, this model will be loaded to continue its training. Once the prepared dataset and model are loaded, the training process is started. N.B. Feel free to cancel the training process at any point, since it will be possible to resume it later on by running the trainer again.

Visualizing Training with TensorBoard

tensorboard --logdir=./logs

Running Predictions

This will build the model using the model builder (cf. model_builder.py) if no model exists yet. If an existing model is found, this model will be loaded. Once the model is loaded, it is used to make predictions on given inputs.

python predictor.py

About

This is a small project which aims to show an example of applied machine learning in Python 3 with the Keras library and its TensorFlow backend to train a neural network model for it to learn to add two integers.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages