Create README.md
Browse files# 🧠 TabTransformer Multitask Model for Churn, Tenure, and LTV Prediction
This model is a multitask `TabTransformer` implemented in PyTorch, designed to perform:
- **Binary classification** for customer **churn**
- **Regression** for customer **tenure**
- **Regression** for customer **LTV (Lifetime Value)**
It is saved as a pickle file: `model.pkl` and includes all custom layers (e.g., positional encoding).
---
## 🧩 Model Architecture
- Tabular input with:
- `x_num`: Numerical features (projected into latent space)
- `x_cat`: Categorical features (embedded + transformer)
- Transformer-based attention over categorical embeddings
- Multi-head output for multitask predictions:
- `Churn`: Sigmoid activation for binary classification
- `Tenure` and `LTV`: Linear regression heads
---
## 🧪 How to Use
### 1. Install Dependencies
```bash
pip install torch pandas