{
  "dataset": "aiml2021/obesity",
  "feature_columns": [
    "Age",
    "Height",
    "Weight",
    "FCVC",
    "NCP",
    "CH2O",
    "FAF",
    "TUE",
    "Gender_Female",
    "Gender_Male",
    "family_history_with_overweight_no",
    "family_history_with_overweight_yes",
    "FAVC_no",
    "FAVC_yes",
    "CAEC_Always",
    "CAEC_Frequently",
    "CAEC_Sometimes",
    "CAEC_no",
    "SMOKE_no",
    "SMOKE_yes",
    "SCC_no",
    "SCC_yes",
    "CALC_Always",
    "CALC_Frequently",
    "CALC_Sometimes",
    "CALC_no",
    "MTRANS_Automobile",
    "MTRANS_Bike",
    "MTRANS_Motorbike",
    "MTRANS_Public_Transportation",
    "MTRANS_Walking"
  ],
  "classes": [
    "Insufficient_Weight",
    "Normal_Weight",
    "Overweight_Level_I",
    "Overweight_Level_II",
    "Obesity_Type_I",
    "Obesity_Type_II",
    "Obesity_Type_III"
  ],
  "n_train": 1688,
  "n_test": 423,
  "regressor": {
    "name": "XGBRegressor",
    "target": "BMI",
    "metrics": {
      "mae": 0.3749128417977634,
      "r2": 0.987534374087826
    },
    "baseline_metrics": {
      "ridge_mae": 0.5687129506345994,
      "ridge_r2": 0.9910858695403162
    }
  },
  "classifier": {
    "name": "XGBClassifier",
    "target": "NObeyesdad",
    "metrics": {
      "accuracy": 0.9574468085106383,
      "macro_f1": 0.9562422503730887
    },
    "baseline_metrics": {
      "logit_accuracy": 0.8628841607565012,
      "logit_macro_f1": 0.8583933672964451
    },
    "per_class": {
      "Insufficient_Weight": {
        "precision": 0.9607843137254902,
        "recall": 0.9074074074074074,
        "f1": 0.9333333333333333,
        "support": 54
      },
      "Normal_Weight": {
        "precision": 0.84375,
        "recall": 0.9310344827586207,
        "f1": 0.8852459016393442,
        "support": 58
      },
      "Overweight_Level_I": {
        "precision": 0.971830985915493,
        "recall": 0.9857142857142858,
        "f1": 0.9787234042553191,
        "support": 70
      },
      "Overweight_Level_II": {
        "precision": 0.9833333333333333,
        "recall": 0.9833333333333333,
        "f1": 0.9833333333333333,
        "support": 60
      },
      "Obesity_Type_I": {
        "precision": 1.0,
        "recall": 0.9846153846153847,
        "f1": 0.9922480620155039,
        "support": 65
      },
      "Obesity_Type_II": {
        "precision": 0.9636363636363636,
        "recall": 0.9137931034482759,
        "f1": 0.9380530973451328,
        "support": 58
      },
      "Obesity_Type_III": {
        "precision": 0.9827586206896551,
        "recall": 0.9827586206896551,
        "f1": 0.9827586206896551,
        "support": 58
      }
    }
  }
}