Create README.md

README.md

@@ -0,0 +1,30 @@
+# 🧠 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