Revenue Prediction Model (AI vs Non-AI Companies)

A machine-learning project that analyzes company revenue, employee trends, and predicts future revenue using LightGBM and Scikit-Learn.

Features

• Revenue prediction using LightGBMRegressor
• AI vs Non-AI revenue & employee growth visualizations
• Model performance metrics (MAE, MSE, RMSE, R²)

Tech Stack

Python, Pandas, NumPy, Matplotlib, Seaborn, Scikit-Learn, LightGBM, Jupyter

---

Setup Instructions

Clone the Repository

git clone https://github.com/curator-me/Revenue-Prediction-model.git
cd Revenue-Prediction-model

Create and activate virtual environment

python3 -m venv venv
source venv/bin/activate          # Mac/Linux
venv\Scripts\activate             # Windows

Install dependencies

pip install -r requirements.txt

Run the notebook

jupyter notebook

---

Project Files

• analysis.ipynb — main notebook
• ai_flagged_company_data.csv — dataset
• README.md — project summary

---