The aim of this project is to serve as a simple example of implementing tensorflow model to be used with the sagemaker-pipeline project.
-
Make sure to have setuptools library installed on python
-
Using version 3.6+ of python
git clone https://github.com/MustafaWaheed91/tf-gamesbiz.git
cd tf-gamesbiz
pip3 install -e .
python3 gamesbiz/train.py
Follow the instructions in sagemaker-pipeline project to see how to use the model in this package to run with SageMaker training pipeline.
- Tensorflow - Open Source Machine Learning framework
- Mustafa Waheed - Data Scientist