experiments.json

{
  "DecisionTree": {
    "params": {
      "max_depth": 10,
      "min_samples_split": 5
    },
    "metrics": {
      "accuracy": 0.87409200968523,
      "precision": 0.7391304347826086,
      "recall": 0.5345911949685535,
      "f1": 0.6204379562043796,
      "roc_auc": 0.8287978652183342
    },
    "time": 3.28193998336792
  },
  "Bagging": {
    "params": {
      "max_samples": 1.0,
      "n_estimators": 100
    },
    "metrics": {
      "accuracy": 0.9164648910411622,
      "precision": 0.9166666666666666,
      "recall": 0.6226415094339622,
      "f1": 0.7415730337078652,
      "roc_auc": 0.9570969232364949
    },
    "time": 14.474764823913574
  },
  "RandomForest": {
    "params": {
      "max_depth": 20,
      "n_estimators": 200
    },
    "metrics": {
      "accuracy": 0.8801452784503632,
      "precision": 0.8488372093023255,
      "recall": 0.4591194968553459,
      "f1": 0.5959183673469388,
      "roc_auc": 0.9396716735971636
    },
    "time": 7.171702861785889
  },
  "AdaBoost": {
    "params": {
      "learning_rate": 1.0,
      "n_estimators": 100
    },
    "metrics": {
      "accuracy": 0.851089588377724,
      "precision": 0.8461538461538461,
      "recall": 0.27672955974842767,
      "f1": 0.41706161137440756,
      "roc_auc": 0.8426400007543398
    },
    "time": 4.963773488998413
  },
  "GradientBoosting": {
    "params": {
      "learning_rate": 0.1,
      "n_estimators": 200
    },
    "metrics": {
      "accuracy": 0.8910411622276029,
      "precision": 0.8709677419354839,
      "recall": 0.5094339622641509,
      "f1": 0.6428571428571429,
      "roc_auc": 0.9011343384911318
    },
    "time": 15.61880111694336
  },
  "XGBoost": {
    "params": {
      "learning_rate": 0.1,
      "n_estimators": 200
    },
    "metrics": {
      "accuracy": 0.9188861985472155,
      "precision": 0.8833333333333333,
      "recall": 0.6666666666666666,
      "f1": 0.7598566308243727,
      "roc_auc": 0.9512130727089285
    },
    "time": 2.7084546089172363
  }
}