Devdit commited on
Commit
aed7890
·
verified ·
1 Parent(s): 4ea1483

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

Files changed (1) hide show
  1. README.md +0 -0
README.md ADDED
File without changes