umer6016 commited on
Commit
a1e09e5
Β·
verified Β·
1 Parent(s): 2aeab56

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +33 -57
README.md CHANGED
@@ -1,57 +1,33 @@
1
- # End-to-End Stock Prediction System
2
-
3
- A comprehensive machine learning system for stock market prediction, featuring data ingestion, processing, model training, and deployment.
4
-
5
- ## Features
6
- - **Data Ingestion**: Fetches daily stock data from Alpha Vantage.
7
- - **Data Processing**: Calculates technical indicators (SMA, RSI, MACD).
8
- - **Machine Learning**:
9
- - **Regression**: Predicts next day's closing price.
10
- - **Classification**: Predicts price direction (Up/Down).
11
- - **Clustering**: Groups market regimes based on volatility.
12
- - **PCA**: Dimensionality reduction for feature analysis.
13
- - **Orchestration**: Prefect workflows for automated pipelines.
14
- - **Validation**: Deepchecks for data integrity and drift detection.
15
- - **Deployment**: Dockerized FastAPI application with Postgres database.
16
- - **CI/CD**: GitHub Actions for testing and deployment.
17
-
18
- ## Tech Stack
19
- - **Language**: Python 3.9
20
- - **Frameworks**: FastAPI, Prefect, Scikit-Learn, Pandas
21
- - **Tools**: Docker, Docker Compose, Deepchecks, Pytest
22
- - **Database**: PostgreSQL
23
-
24
- ## Quick Start
25
-
26
- ### Prerequisites
27
- - Docker & Docker Compose
28
- - Alpha Vantage API Key (set in `.env`)
29
-
30
- ### Installation
31
- 1. Clone the repository.
32
- 2. Create a `.env` file:
33
- ```bash
34
- cp .env.example .env
35
- # Edit .env with your API key
36
- ```
37
- 3. Build and start services:
38
- ```bash
39
- docker-compose up --build -d
40
- ```
41
-
42
- ### Usage
43
- - **API Documentation**: [http://localhost:8000/docs](http://localhost:8000/docs)
44
- - **Prefect UI**: [http://localhost:4200](http://localhost:4200)
45
- - **Health Check**: [http://localhost:8000/health](http://localhost:8000/health)
46
-
47
- ### Running Tests
48
- ```bash
49
- pip install -e .[dev]
50
- python -m pytest tests/
51
- ```
52
-
53
- ### Training Models
54
- To train models manually:
55
- ```bash
56
- python src/orchestration/flows.py
57
- ```
 
1
+ ---
2
+ title: AI Stock Prediction System
3
+ emoji: πŸ“ˆ
4
+ colorFrom: blue
5
+ colorTo: green
6
+ sdk: docker
7
+ pinned: false
8
+ app_port: 7860
9
+ ---
10
+
11
+ # πŸ“ˆ AI Stock Prediction & Analysis System
12
+
13
+ An End-to-End Machine Learning System for predicting stock prices and analyzing market regimes.
14
+
15
+ ## πŸš€ Features
16
+
17
+ * **Ensemble Models:** Combines Linear Regression, Random Forest, and SVM for robust predictions.
18
+ * **Live Data:** Fetches real-time price data from Alpha Vantage.
19
+ * **Market Analysis:** Unsupervised Learning (Clustering & PCA) to detect market volatility regimes.
20
+ * **Automated Pipeline:** Built with Prefect for orchestration and GitHub Actions for CI/CD.
21
+
22
+ ## πŸ› οΈ Tech Stack
23
+
24
+ * **Frontend:** Streamlit
25
+ * **Backend:** FastAPI / Prefect
26
+ * **ML:** Scikit-Learn, Pandas, Numpy
27
+ * **DevOps:** Docker, GitHub Actions
28
+
29
+ ## πŸ“Š How to Use
30
+
31
+ 1. Select a stock from the sidebar (e.g., AAPL, NVDA).
32
+ 2. View the Real-Time Metrics and AI Predictions.
33
+ 3. Switch tabs to explore the "Market Regime" visualization.