This project contains machine learning models and fine-tuning experiments.
fine_tuning_results/
: Contains the results of hyperparameter search experiments.hausing_widedeep_model/
: Contains the saved model files for a Wide & Deep model for housing data.Neural_Network_fine-tuning.ipynb
: Jupyter notebook for fine-tuning neural networks.TF_models.ipynb
: Jupyter notebook for TensorFlow models.
hausing_widedeep_model/fingerprint.pb
: TensorFlow model fingerprint file.hausing_widedeep_model/keras_metadata.pb
: Keras metadata for the TensorFlow model.
Open the Jupyter notebooks (Neural_Network_fine-tuning.ipynb
, TF_models.ipynb
) to view the model training and evaluation process.
- Python 3
- TensorFlow
- Keras
- Jupyter
- Clone this repository.
- Install the required packages.
- Open the Jupyter notebooks to run the models.
Please read CONTRIBUTING.md for details on our code of conduct, and the process for submitting pull requests to us.
This project is licensed under the MIT License - see the LICENSE.md file for details