Skip to content

Commit

Permalink
Added docker file and updated README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
Shakleen committed Oct 15, 2024
1 parent 6c2ca4d commit b480fd5
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 1 deletion.
6 changes: 6 additions & 0 deletions DOCKERFILE
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
FROM python:3.11
COPY . /app
WORKDIR /app
RUN pip install .
EXPOSE $PORT
CMD gunicorn --workers=1 --bind 0.0.0.0:$PORT app:app
3 changes: 2 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,8 @@ This project is a data-driven system designed to predict hourly bike and dock de
![MLflow](https://img.shields.io/badge/MLflow-Tracking%20Experiment-blue)
![Flask](https://img.shields.io/badge/Flask-Web%20Framework-lightgrey)
![PyTest](https://img.shields.io/badge/PyTest-Testing-green)
![Azure](https://img.shields.io/badge/Azure-Cloud_Service-blue)
![Docker](https://img.shields.io/badge/Docker-Containerization-blue)

<center><img src="Diagrams/AppDemo.gif"/></center>

Expand Down Expand Up @@ -78,7 +80,6 @@ The system's machine learning pipeline is built using Spark MLlib for distribute

* **Random Forest** - Decision-tree based ensemble model.
* **Gradient Boosted Trees** - Advanced ensemble model for better accuracy.
* **Temporal Graph Neural Networks** - Time-series forecasting model for graph data.

All models are trained on the Gold Delta Lake and evaluated based on their prediction accuracy for hourly bike and dock demand.

Expand Down

0 comments on commit b480fd5

Please sign in to comment.