# vaibhav8831/ai-impact-jobs-layoff-prediction_model architecture

> Architecture diagram of the GitHub repository vaibhav8831/ai-impact-jobs-layoff-prediction_model, generated by GitDiagram from its file tree, README and a sample of source files (last updated 2026-09-28). It is an AI-made overview: check details against the source.

- Interactive diagram: https://gitdiagram.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model
- Repository: https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model

## Overview

This repository is a batch machine-learning project for predicting job layoff risk. Running `main.py` loads the bundled CSV, prepares and splits the data, trains Random Forest and XGBoost pipelines, and evaluates both prediction sets with weighted classification metrics. The graph reflects the sampled Python workflow; the README provides no further product details, and `app.py` is empty.

Main entry points: [`main.py`](https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/main.py) (Workflow Entry), [`src/data_ingation.py`](https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/src/data_ingation.py) (CSV Loader), [`src/data_modelbuilding.py`](https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/src/data_modelbuilding.py) (Model Training), [`data/ai-impact-jobs-layoff-risk-dataset.csv`](https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/data/ai-impact-jobs-layoff-risk-dataset.csv) (Risk Dataset)

## Diagram (Mermaid)

```mermaid
flowchart TD

subgraph group_input["Data Input"]
  node_csv["Risk Dataset"]
  node_loader["CSV Loader<br/>[data_ingation.py]"]
end

subgraph group_modeling["Modeling"]
  node_main["Workflow Entry<br/>[main.py]"]
  node_preprocess["Data Preparation<br/>[data_preproces.py]"]
  node_trainer["Model Training"]
end

subgraph group_evaluation["Evaluation"]
  node_evaluator["Metric Evaluation<br/>[modelevolation.py]"]
end

node_runner(("Run Script"))

node_runner -->|"runs"| node_main
node_main -->|"calls"| node_loader
node_loader -->|"reads"| node_csv
node_loader -->|"returns data"| node_main
node_main -->|"calls"| node_trainer
node_trainer -->|"calls"| node_preprocess
node_trainer -->|"returns predictions"| node_main
node_main -->|"evaluates predictions"| node_evaluator

click node_main "https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/main.py"
click node_csv "https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/data/ai-impact-jobs-layoff-risk-dataset.csv"
click node_loader "https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/src/data_ingation.py"
click node_preprocess "https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/src/data_preproces.py"
click node_trainer "https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/src/data_modelbuilding.py"
click node_evaluator "https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/src/modelevolation.py"

classDef toneNeutral fill:#f8fafc,stroke:#334155,stroke-width:1.5px,color:#0f172a
classDef toneBlue fill:#dbeafe,stroke:#2563eb,stroke-width:1.5px,color:#172554
classDef toneAmber fill:#fef3c7,stroke:#d97706,stroke-width:1.5px,color:#78350f
classDef toneMint fill:#dcfce7,stroke:#16a34a,stroke-width:1.5px,color:#14532d
classDef toneRose fill:#ffe4e6,stroke:#e11d48,stroke-width:1.5px,color:#881337
classDef toneIndigo fill:#e0e7ff,stroke:#4f46e5,stroke-width:1.5px,color:#312e81
classDef toneTeal fill:#ccfbf1,stroke:#0f766e,stroke-width:1.5px,color:#134e4a
class node_csv,node_loader toneBlue
class node_main,node_preprocess,node_trainer toneAmber
class node_evaluator toneMint
class node_runner toneIndigo
```

## Components

### Data Input

- [Risk Dataset](https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/data/ai-impact-jobs-layoff-risk-dataset.csv): `data/ai-impact-jobs-layoff-risk-dataset.csv`
- [CSV Loader](https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/src/data_ingation.py): `src/data_ingation.py`

### Modeling

- [Workflow Entry](https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/main.py): `main.py`
- [Data Preparation](https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/src/data_preproces.py): `src/data_preproces.py`
- [Model Training](https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/src/data_modelbuilding.py): `src/data_modelbuilding.py`

### Evaluation

- [Metric Evaluation](https://github.com/vaibhav8831/ai-impact-jobs-layoff-prediction_model/blob/main/src/modelevolation.py): `src/modelevolation.py`

### Other components

- **Run Script**

## Connections

- Run Script → Workflow Entry: runs
- Workflow Entry → CSV Loader: calls
- CSV Loader → Risk Dataset: reads
- CSV Loader → Workflow Entry: returns data
- Workflow Entry → Model Training: calls
- Model Training → Data Preparation: calls
- Model Training → Workflow Entry: returns predictions
- Workflow Entry → Metric Evaluation: evaluates predictions