sathishaiuse/wellness-tourism-model-best

Wellness Tourism — Best Model

Model repository: sathishaiuse/wellness-tourism-model-best

Trained with: Optuna hyperparameter tuning and MLflow tracking.

Evaluation metrics (test set)

{
  "accuracy": 0.9055690072639225,
  "precision": 0.945054945054945,
  "recall": 0.5408805031446541,
  "f1": 0.688,
  "roc_auc": 0.9681762892138838,
  "confusion_matrix": [
    [
      662,
      5
    ],
    [
      73,
      86
    ]
  ],
  "classification_report": {
    "0": {
      "precision": 0.9006802721088435,
      "recall": 0.992503748125937,
      "f1-score": 0.9443651925820257,
      "support": 667.0
    },
    "1": {
      "precision": 0.945054945054945,
      "recall": 0.5408805031446541,
      "f1-score": 0.688,
      "support": 159.0
    },
    "accuracy": 0.9055690072639225,
    "macro avg": {
      "precision": 0.9228676085818943,
      "recall": 0.7666921256352955,
      "f1-score": 0.8161825962910128,
      "support": 826.0
    },
    "weighted avg": {
      "precision": 0.9092221280391464,
      "recall": 0.9055690072639225,
      "f1-score": 0.8950164448573985,
      "support": 826.0
    }
  }
}

Usage

Download best_pipeline.joblib and load with joblib.load().