{ "cells": [ { "cell_type": "markdown", "metadata": {}, "source": [ "# Example: Multi-metric runs\n", "----------------------------\n", "\n", "This example shows how to evaluate an atom's pipeline on multiple metrics.\n", "\n", "Import the breast cancer dataset from [sklearn.datasets](https://scikit-learn.org/stable/datasets/index.html#wine-dataset). This is a small and easy to train dataset whose goal is to predict whether a patient has breast cancer or not." ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "## Load the data" ] }, { "cell_type": "code", "execution_count": 1, "metadata": {}, "outputs": [], "source": [ "# Import packages\n", "import pandas as pd\n", "from atom import ATOMRegressor" ] }, { "cell_type": "code", "execution_count": 2, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "\n", "\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "
SexLengthDiameterHeightWhole weightShucked weightViscera weightShell weightRings
0M0.4550.3650.0950.51400.22450.10100.15015
1M0.3500.2650.0900.22550.09950.04850.0707
2F0.5300.4200.1350.67700.25650.14150.2109
3M0.4400.3650.1250.51600.21550.11400.15510
4I0.3300.2550.0800.20500.08950.03950.0557
\n", "
" ], "text/plain": [ " Sex Length Diameter Height Whole weight Shucked weight Viscera weight \\\n", "0 M 0.455 0.365 0.095 0.5140 0.2245 0.1010 \n", "1 M 0.350 0.265 0.090 0.2255 0.0995 0.0485 \n", "2 F 0.530 0.420 0.135 0.6770 0.2565 0.1415 \n", "3 M 0.440 0.365 0.125 0.5160 0.2155 0.1140 \n", "4 I 0.330 0.255 0.080 0.2050 0.0895 0.0395 \n", "\n", " Shell weight Rings \n", "0 0.150 15 \n", "1 0.070 7 \n", "2 0.210 9 \n", "3 0.155 10 \n", "4 0.055 7 " ] }, "execution_count": 2, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# Load data\n", "X = pd.read_csv(\"./datasets/abalone.csv\")\n", "\n", "# Let's have a look\n", "X.head()" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "## Run the pipeline" ] }, { "cell_type": "code", "execution_count": 3, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "<< ================== ATOM ================== >>\n", "\n", "Configuration ==================== >>\n", "Algorithm task: Regression.\n", "\n", "Dataset stats ==================== >>\n", "Shape: (4177, 9)\n", "Train set size: 3342\n", "Test set size: 835\n", "-------------------------------------\n", "Memory: 300.88 kB\n", "Scaled: False\n", "Categorical features: 1 (12.5%)\n", "Outlier values: 189 (0.6%)\n", "\n" ] } ], "source": [ "atom = ATOMRegressor(X, n_jobs=1, verbose=2, random_state=1)" ] }, { "cell_type": "code", "execution_count": 4, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Fitting Encoder...\n", "Encoding categorical columns...\n", " --> OneHot-encoding feature Sex. Contains 3 classes.\n" ] } ], "source": [ "atom.encode()" ] }, { "cell_type": "code", "execution_count": 5, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "\n", "Training ========================= >>\n", "Models: lSVM, hGBM\n", "Metric: r2, rmse\n", "\n", "\n", "Running hyperparameter tuning for LinearSVM...\n", "| trial | loss | C | dual | r2 | best_r2 | rmse | best_rmse | time_trial | time_ht | state |\n", "| ----- | ----------------------- | ------- | ------- | ------- | ------- | ------- | --------- | ---------- | ------- | -------- |\n", "| 0 | squared_epsilon_insen.. | 0.001 | True | 0.2887 | 0.2887 | -2.6528 | -2.6528 | 0.081s | 0.081s | COMPLETE |\n", "| 1 | squared_epsilon_insen.. | 0.0534 | False | 0.3862 | 0.3862 | -2.5926 | -2.5926 | 0.078s | 0.159s | COMPLETE |\n", "| 2 | squared_epsilon_insen.. | 0.0105 | True | 0.433 | 0.433 | -2.4084 | -2.4084 | 0.070s | 0.229s | COMPLETE |\n", "| 3 | epsilon_insensitive | 0.6215 | True | 0.4022 | 0.433 | -2.5251 | -2.4084 | 0.064s | 0.293s | COMPLETE |\n", "| 4 | squared_epsilon_insen.. | 0.0369 | False | 0.4057 | 0.433 | -2.5477 | -2.4084 | 0.058s | 0.351s | COMPLETE |\n", "| 5 | epsilon_insensitive | 0.0016 | True | -1.5344 | 0.433 | -5.0102 | -2.4084 | 0.052s | 0.403s | COMPLETE |\n", "| 6 | squared_epsilon_insen.. | 61.5811 | False | 0.4354 | 0.4354 | -2.3845 | -2.3845 | 0.047s | 0.450s | COMPLETE |\n", "| 7 | squared_epsilon_insen.. | 14.898 | False | 0.4925 | 0.4925 | -2.2628 | -2.2628 | 0.050s | 0.500s | COMPLETE |\n", "| 8 | epsilon_insensitive | 0.0252 | True | 0.3695 | 0.4925 | -2.6178 | -2.2628 | 0.048s | 0.548s | COMPLETE |\n", "| 9 | squared_epsilon_insen.. | 0.0294 | True | 0.4767 | 0.4925 | -2.3896 | -2.2628 | 0.055s | 0.604s | COMPLETE |\n", "Hyperparameter tuning ---------------------------\n", "Best trial --> 7\n", "Best parameters:\n", " --> loss: squared_epsilon_insensitive\n", " --> C: 14.898\n", " --> dual: False\n", "Best evaluation --> r2: 0.4925 rmse: -2.2628\n", "Time elapsed: 0.604s\n", "Fit ---------------------------------------------\n", "Train evaluation --> r2: 0.4592 rmse: -2.3795\n", "Test evaluation --> r2: 0.4584 rmse: -2.3369\n", "Time elapsed: 0.112s\n", "Bootstrap ---------------------------------------\n", "Evaluation --> r2: 0.4577 ± 0.002 rmse: -2.3384 ± 0.0043\n", "Time elapsed: 0.131s\n", "-------------------------------------------------\n", "Time: 0.847s\n", "\n", "\n", "Running hyperparameter tuning for HistGradientBoosting...\n", "| trial | loss | quantile | learning_rate | max_iter | max_leaf_nodes | max_depth | min_samples_leaf | l2_regularization | r2 | best_r2 | rmse | best_rmse | time_trial | time_ht | state |\n", "| ----- | --------- | -------- | ------------- | -------- | -------------- | --------- | ---------------- | ----------------- | ------- | ------- | ------- | --------- | ---------- | ------- | -------- |\n", "| 0 | absolut.. | 0.1 | 0.0236 | 180 | 26 | 12 | 11 | 0.0 | 0.5373 | 0.5373 | -2.1398 | -2.1398 | 0.933s | 0.933s | COMPLETE |\n", "| 1 | gamma | 0.5 | 0.242 | 160 | 38 | 3 | 20 | 0.0 | 0.574 | 0.574 | -2.1598 | -2.1398 | 0.166s | 1.099s | COMPLETE |\n", "| 2 | quantile | 0.4 | 0.2448 | 210 | 12 | 3 | 25 | 0.3 | 0.4714 | 0.574 | -2.3253 | -2.1398 | 0.474s | 1.573s | COMPLETE |\n", "| 3 | quantile | 0.6 | 0.017 | 480 | 28 | 16 | 13 | 0.1 | 0.5712 | 0.574 | -2.1385 | -2.1385 | 3.667s | 5.241s | COMPLETE |\n", "| 4 | squared.. | 1.0 | 0.2649 | 70 | 10 | 10 | 28 | 0.8 | 0.5561 | 0.574 | -2.2019 | -2.1385 | 0.167s | 5.408s | COMPLETE |\n", "| 5 | squared.. | 0.1 | 0.0283 | 360 | 32 | 9 | 11 | 0.5 | 0.5464 | 0.574 | -2.1197 | -2.1197 | 1.312s | 6.720s | COMPLETE |\n", "| 6 | quantile | 0.4 | 0.1264 | 380 | 37 | 12 | 29 | 1.0 | 0.4417 | 0.574 | -2.3712 | -2.1197 | 3.378s | 10.098s | COMPLETE |\n", "| 7 | gamma | 0.6 | 0.678 | 330 | 25 | 6 | 12 | 0.8 | 0.4299 | 0.574 | -2.3984 | -2.1197 | 0.807s | 10.904s | COMPLETE |\n", "| 8 | absolut.. | 0.9 | 0.0831 | 280 | 42 | 16 | 10 | 1.0 | 0.5242 | 0.574 | -2.2742 | -2.1197 | 2.341s | 13.246s | COMPLETE |\n", "| 9 | absolut.. | 0.6 | 0.0373 | 300 | 40 | 13 | 17 | 0.8 | 0.5685 | 0.574 | -2.17 | -2.1197 | 2.032s | 15.277s | COMPLETE |\n", "Hyperparameter tuning ---------------------------\n", "Best trial --> 5\n", "Best parameters:\n", " --> loss: squared_error\n", " --> quantile: 0.1\n", " --> learning_rate: 0.0283\n", " --> max_iter: 360\n", " --> max_leaf_nodes: 32\n", " --> max_depth: 9\n", " --> min_samples_leaf: 11\n", " --> l2_regularization: 0.5\n", "Best evaluation --> r2: 0.5464 rmse: -2.1197\n", "Time elapsed: 15.277s\n", "Fit ---------------------------------------------\n", "Train evaluation --> r2: 0.7959 rmse: -1.4619\n", "Test evaluation --> r2: 0.5479 rmse: -2.135\n", "Time elapsed: 1.311s\n", "Bootstrap ---------------------------------------\n", "Evaluation --> r2: 0.5259 ± 0.0154 rmse: -2.1862 ± 0.0353\n", "Time elapsed: 4.073s\n", "-------------------------------------------------\n", "Time: 20.661s\n", "\n", "\n", "Final results ==================== >>\n", "Total time: 22.907s\n", "-------------------------------------\n", "LinearSVM --> r2: 0.4577 ± 0.002 rmse: -2.3384 ± 0.0043\n", "HistGradientBoosting --> r2: 0.5259 ± 0.0154 rmse: -2.1862 ± 0.0353 ~ !\n" ] } ], "source": [ "# For every step of the BO, both metrics are calculated,\n", "# but only the first is used for optimization!\n", "atom.run(\n", " models=[\"lsvm\", \"hGBM\"],\n", " metric=(\"r2\", \"rmse\"),\n", " n_trials=10,\n", " n_bootstrap=6,\n", ")" ] }, { "cell_type": "code", "execution_count": 6, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Applying cross-validation...\n" ] }, { "data": { "text/html": [ "
\n", "\n", "\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "
train_r2test_r2train_rmsetest_rmsetime (s)
00.7960380.542475-1.453147-2.1957790.640582
10.7949540.540335-1.457709-2.1963910.644585
20.7907220.505952-1.492522-2.1533920.672611
30.7853170.580703-1.474827-2.1899020.684622
40.7958720.547938-1.461929-2.1350220.741673
mean0.7925810.543480-1.468027-2.1740970.676815
std0.0041140.0237690.0142220.0251890.036433
\n", "
" ], "text/plain": [ " train_r2 test_r2 train_rmse test_rmse time (s)\n", "0 0.796038 0.542475 -1.453147 -2.195779 0.640582\n", "1 0.794954 0.540335 -1.457709 -2.196391 0.644585\n", "2 0.790722 0.505952 -1.492522 -2.153392 0.672611\n", "3 0.785317 0.580703 -1.474827 -2.189902 0.684622\n", "4 0.795872 0.547938 -1.461929 -2.135022 0.741673\n", "mean 0.792581 0.543480 -1.468027 -2.174097 0.676815\n", "std 0.004114 0.023769 0.014222 0.025189 0.036433" ] }, "execution_count": 6, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# Check the robustness of the pipeline using cross-validation\n", "atom.winner.cross_validate()" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "## Analyze the results" ] }, { "cell_type": "code", "execution_count": 7, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "\n", "\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "
r2_htr2_trainr2_testrmse_htrmse_trainrmse_test
lSVM0.4925300.45830.4552-2.262754-2.3815-2.3439
hGBM0.5463680.79590.5479-2.119672-1.4619-2.1350
\n", "
" ], "text/plain": [ " r2_ht r2_train r2_test rmse_ht rmse_train rmse_test\n", "lSVM 0.492530 0.4583 0.4552 -2.262754 -2.3815 -2.3439\n", "hGBM 0.546368 0.7959 0.5479 -2.119672 -1.4619 -2.1350" ] }, "execution_count": 7, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# The columns in the results dataframe contain one for each metric\n", "atom.results[[\"r2_ht\", \"r2_train\", \"r2_test\", \"rmse_ht\", \"rmse_train\", \"rmse_test\"]]" ] }, { "cell_type": "code", "execution_count": 8, "metadata": {}, "outputs": [ { "data": { "text/html": [ " \n", " " ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.plotly.v1+json": { "config": { "plotlyServerURL": "https://plotly.com" }, "data": [ { "legendgroup": "lSVM", "legendgrouptitle": { "font": { "size": 16 }, "text": "lSVM" }, "line": { "color": "rgb(0, 98, 98)", "dash": "solid", "width": 2 }, "marker": { "color": "rgb(0, 98, 98)", "line": { "color": "rgba(255, 255, 255, 0.9)", "width": 1 }, "size": [ 8, 8, 8, 8, 8, 8, 8, 12, 8, 8 ], "symbol": [ "circle", "circle", "circle", "circle", "circle", "circle", "circle", "star", "circle", "circle" ] }, "mode": "lines+markers", "name": "r2", "showlegend": true, "type": "scatter", "x": [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 ], "xaxis": "x2", "y": [ 0.28873632968867413, 0.38618502821310496, 0.432972702836662, 0.4021724449668481, 0.40574506271683086, -1.534383966493194, 0.4353899817981043, 0.4925303455788521, 0.3695277003837695, 0.4766691395610483 ], "yaxis": "y" }, { "hovertemplate": "(%{x}, %{y})lSVM - r2", "legendgroup": "lSVM", "legendgrouptitle": { "font": { "size": 16 }, "text": "lSVM" }, "line": { "color": "rgb(0, 98, 98)", "dash": "solid", "width": 2 }, "marker": { "color": "rgb(0, 98, 98)", "line": { "color": "rgba(255, 255, 255, 0.9)", "width": 1 }, "size": 8, "symbol": "circle" }, "mode": "lines+markers", "name": "r2", "showlegend": false, "type": "scatter", "x": [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 ], "xaxis": "x2", "y": [ null, 0.09744869852443083, 0.04678767462355704, -0.030800257869813885, 0.0035726177499827427, -1.9401290292100248, 1.9697739482912984, 0.05714036378074783, -0.12300264519508264, 0.10714143917727881 ], "yaxis": "y2" }, { "legendgroup": "hGBM", "legendgrouptitle": { "font": { "size": 16 }, "text": "hGBM" }, "line": { "color": "rgb(56, 166, 165)", "dash": "solid", "width": 2 }, "marker": { "color": "rgb(56, 166, 165)", "line": { "color": "rgba(255, 255, 255, 0.9)", "width": 1 }, "size": [ 8, 8, 8, 8, 8, 12, 8, 8, 8, 8 ], "symbol": [ "circle", "circle", "circle", "circle", "circle", "star", "circle", "circle", "circle", "circle" ] }, "mode": "lines+markers", "name": "r2", "showlegend": true, "type": "scatter", "x": [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 ], "xaxis": "x2", "y": [ 0.5372556317864228, 0.5740339542456956, 0.47142836502638574, 0.5712329261259416, 0.5560963591921516, 0.5463677759098945, 0.4416681886675389, 0.42986288303461084, 0.5241627607090207, 0.5684504515546619 ], "yaxis": "y" }, { "hovertemplate": "(%{x}, %{y})hGBM - r2", "legendgroup": "hGBM", "legendgrouptitle": { "font": { "size": 16 }, "text": "hGBM" }, "line": { "color": "rgb(56, 166, 165)", "dash": "solid", "width": 2 }, "marker": { "color": "rgb(56, 166, 165)", "line": { "color": "rgba(255, 255, 255, 0.9)", "width": 1 }, "size": 8, "symbol": "circle" }, "mode": "lines+markers", "name": "r2", "showlegend": false, "type": "scatter", "x": [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 ], "xaxis": "x2", "y": [ null, 0.036778322459272816, -0.10260558921930985, 0.09980456109955582, -0.015136566933789952, -0.00972858328225712, -0.10469958724235562, -0.011805305632928031, 0.09429987767440984, 0.044287690845641214 ], "yaxis": "y2" }, { "legendgroup": "lSVM", "legendgrouptitle": { "font": { "size": 16 }, "text": "lSVM" }, "line": { "color": "rgb(0, 98, 98)", "dash": "dashdot", "width": 2 }, "marker": { "color": "rgb(0, 98, 98)", "line": { "color": "rgba(255, 255, 255, 0.9)", "width": 1 }, "size": [ 8, 8, 8, 8, 8, 8, 8, 12, 8, 8 ], "symbol": [ "circle", "circle", "circle", "circle", "circle", "circle", "circle", "star", "circle", "circle" ] }, "mode": "lines+markers", "name": "rmse", "showlegend": true, "type": "scatter", "x": [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 ], "xaxis": "x4", "y": [ -2.652832750951567, -2.5926015099951765, -2.4083996657900633, -2.525147605339932, -2.547687907299415, -5.0101759500263086, -2.3844964091771974, -2.262753922393612, -2.6178084479643786, -2.3896144120353857 ], "yaxis": "y3" }, { "hovertemplate": "(%{x}, %{y})lSVM - rmse", "legendgroup": "lSVM", "legendgrouptitle": { "font": { "size": 16 }, "text": "lSVM" }, "line": { "color": "rgb(0, 98, 98)", "dash": "dashdot", "width": 2 }, "marker": { "color": "rgb(0, 98, 98)", "line": { "color": "rgba(255, 255, 255, 0.9)", "width": 1 }, "size": 8, "symbol": "circle" }, "mode": "lines+markers", "name": "rmse", "showlegend": false, "type": "scatter", "x": [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 ], "xaxis": "x4", "y": [ null, 0.06023124095639032, 0.1842018442051132, -0.11674793954986873, -0.022540301959482978, -2.4624880427268936, 2.625679540849111, 0.12174248678358524, -0.35505452557076644, 0.2281940359289929 ], "yaxis": "y4" }, { "legendgroup": "hGBM", "legendgrouptitle": { "font": { "size": 16 }, "text": "hGBM" }, "line": { "color": "rgb(56, 166, 165)", "dash": "dashdot", "width": 2 }, "marker": { "color": "rgb(56, 166, 165)", "line": { "color": "rgba(255, 255, 255, 0.9)", "width": 1 }, "size": [ 8, 8, 8, 8, 8, 12, 8, 8, 8, 8 ], "symbol": [ "circle", "circle", "circle", "circle", "circle", "star", "circle", "circle", "circle", "circle" ] }, "mode": "lines+markers", "name": "rmse", "showlegend": true, "type": "scatter", "x": [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 ], "xaxis": "x4", "y": [ -2.139760313150705, -2.1597551695224046, -2.3252973771629937, -2.138501301350621, -2.201933403862386, -2.1196722802537042, -2.3712020908397347, -2.3984018212691107, -2.274228372666243, -2.1699762938781455 ], "yaxis": "y3" }, { "hovertemplate": "(%{x}, %{y})hGBM - rmse", "legendgroup": "hGBM", "legendgrouptitle": { "font": { "size": 16 }, "text": "hGBM" }, "line": { "color": "rgb(56, 166, 165)", "dash": "dashdot", "width": 2 }, "marker": { "color": "rgb(56, 166, 165)", "line": { "color": "rgba(255, 255, 255, 0.9)", "width": 1 }, "size": 8, "symbol": "circle" }, "mode": "lines+markers", "name": "rmse", "showlegend": false, "type": "scatter", "x": [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 ], "xaxis": "x4", "y": [ null, -0.019994856371699843, -0.16554220764058902, 0.18679607581237256, -0.06343210251176501, 0.08226112360868187, -0.2515298105860304, -0.027199730429376068, 0.12417344860286761, 0.10425207878809761 ], "yaxis": "y4" } ], "layout": { "annotations": [ { "font": { "size": 20 }, "showarrow": false, "text": "Hyperparameter tuning performance for R2", "x": 0.225, "xanchor": "center", "xref": "paper", "y": 1.005, "yanchor": "bottom", "yref": "paper" }, { "font": { "size": 20 }, "showarrow": false, "text": "Hyperparameter tuning performance for RMSE", "x": 0.775, "xanchor": "center", "xref": "paper", "y": 1.005, "yanchor": "bottom", "yref": "paper" } ], "font": { "size": 12 }, "height": 600, "hoverlabel": { "font": { "size": 16 } }, "hovermode": "x unified", "legend": { "bgcolor": "rgba(255, 255, 255, 0.5)", "font": { "size": 16 }, "groupclick": "togglegroup", "traceorder": "grouped" }, "margin": { "b": 50, "l": 50, "pad": 0, "r": 0, "t": 59 }, "showlegend": true, "template": { "data": { "bar": [ { "error_x": { "color": "#2a3f5f" }, "error_y": { "color": "#2a3f5f" }, "marker": { "line": { "color": "#E5ECF6", "width": 0.5 }, "pattern": { "fillmode": "overlay", "size": 10, "solidity": 0.2 } }, "type": "bar" } ], "barpolar": [ { "marker": { "line": { "color": "#E5ECF6", "width": 0.5 }, "pattern": { "fillmode": "overlay", "size": 10, "solidity": 0.2 } }, "type": "barpolar" } ], "carpet": [ { "aaxis": { "endlinecolor": "#2a3f5f", "gridcolor": "white", "linecolor": "white", "minorgridcolor": "white", "startlinecolor": "#2a3f5f" }, "baxis": { "endlinecolor": "#2a3f5f", "gridcolor": "white", "linecolor": "white", "minorgridcolor": "white", "startlinecolor": "#2a3f5f" }, "type": "carpet" } ], "choropleth": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "type": "choropleth" } ], "contour": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "colorscale": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ], "type": "contour" } ], "contourcarpet": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "type": "contourcarpet" } ], "heatmap": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "colorscale": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ], "type": "heatmap" } ], "heatmapgl": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "colorscale": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ], "type": "heatmapgl" } ], "histogram": [ { "marker": { "pattern": { "fillmode": "overlay", "size": 10, "solidity": 0.2 } }, "type": "histogram" } ], "histogram2d": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "colorscale": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ], "type": "histogram2d" } ], "histogram2dcontour": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "colorscale": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ], "type": "histogram2dcontour" } ], "mesh3d": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "type": "mesh3d" } ], "parcoords": [ { "line": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "parcoords" } ], "pie": [ { "automargin": true, "type": "pie" } ], "scatter": [ { "fillpattern": { "fillmode": "overlay", "size": 10, "solidity": 0.2 }, "type": "scatter" } ], "scatter3d": [ { "line": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scatter3d" } ], "scattercarpet": [ { "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scattercarpet" } ], "scattergeo": [ { "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scattergeo" } ], "scattergl": [ { "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scattergl" } ], "scattermapbox": [ { "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scattermapbox" } ], "scatterpolar": [ { "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scatterpolar" } ], "scatterpolargl": [ { "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scatterpolargl" } ], "scatterternary": [ { "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scatterternary" } ], "surface": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "colorscale": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ], "type": "surface" } ], "table": [ { "cells": { "fill": { "color": "#EBF0F8" }, "line": { "color": "white" } }, "header": { "fill": { "color": "#C8D4E3" }, "line": { "color": "white" } }, "type": "table" } ] }, "layout": { "annotationdefaults": { "arrowcolor": "#2a3f5f", "arrowhead": 0, "arrowwidth": 1 }, "autotypenumbers": "strict", "coloraxis": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "colorscale": { "diverging": [ [ 0, "#8e0152" ], [ 0.1, "#c51b7d" ], [ 0.2, "#de77ae" ], [ 0.3, "#f1b6da" ], [ 0.4, "#fde0ef" ], [ 0.5, "#f7f7f7" ], [ 0.6, "#e6f5d0" ], [ 0.7, "#b8e186" ], [ 0.8, "#7fbc41" ], [ 0.9, "#4d9221" ], [ 1, "#276419" ] ], "sequential": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ], "sequentialminus": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ] }, "colorway": [ "#636efa", "#EF553B", "#00cc96", "#ab63fa", "#FFA15A", "#19d3f3", "#FF6692", "#B6E880", "#FF97FF", "#FECB52" ], "font": { "color": "#2a3f5f" }, "geo": { "bgcolor": "white", "lakecolor": "white", "landcolor": "#E5ECF6", "showlakes": true, "showland": true, "subunitcolor": "white" }, "hoverlabel": { "align": "left" }, "hovermode": "closest", "mapbox": { "style": "light" }, "paper_bgcolor": "white", "plot_bgcolor": "#E5ECF6", "polar": { "angularaxis": { "gridcolor": "white", "linecolor": "white", "ticks": "" }, "bgcolor": "#E5ECF6", "radialaxis": { "gridcolor": "white", "linecolor": "white", "ticks": "" } }, "scene": { "xaxis": { "backgroundcolor": "#E5ECF6", "gridcolor": "white", "gridwidth": 2, "linecolor": "white", "showbackground": true, "ticks": "", "zerolinecolor": "white" }, "yaxis": { "backgroundcolor": "#E5ECF6", "gridcolor": "white", "gridwidth": 2, "linecolor": "white", "showbackground": true, "ticks": "", "zerolinecolor": "white" }, "zaxis": { "backgroundcolor": "#E5ECF6", "gridcolor": "white", "gridwidth": 2, "linecolor": "white", "showbackground": true, "ticks": "", "zerolinecolor": "white" } }, "shapedefaults": { "line": { "color": "#2a3f5f" } }, "ternary": { "aaxis": { "gridcolor": "white", "linecolor": "white", "ticks": "" }, "baxis": { "gridcolor": "white", "linecolor": "white", "ticks": "" }, "bgcolor": "#E5ECF6", "caxis": { "gridcolor": "white", "linecolor": "white", "ticks": "" } }, "title": { "x": 0.05 }, "xaxis": { "automargin": true, "gridcolor": "white", "linecolor": "white", "ticks": "", "title": { "standoff": 15 }, "zerolinecolor": "white", "zerolinewidth": 2 }, "yaxis": { "automargin": true, "gridcolor": "white", "linecolor": "white", "ticks": "", "title": { "standoff": 15 }, "zerolinecolor": "white", "zerolinewidth": 2 } } }, "title": { "font": { "size": 24 }, "pad": { "b": 15, "t": 15 }, "x": 0.5, "xanchor": "center", "xref": "paper", "y": 1, "yanchor": "top" }, "width": 1250, "xaxis": { "anchor": "y", "automargin": true, "domain": [ 0, 0.45 ], "showticklabels": false, "title": { "font": { "size": 16 } } }, "xaxis2": { "anchor": "y2", "automargin": true, "autorange": true, "domain": [ 0, 0.45 ], "range": [ -0.6164347674215521, 9.616434767421552 ], "title": { "font": { "size": 16 }, "text": "Trial" }, "type": "linear" }, "xaxis3": { "anchor": "y3", "automargin": true, "domain": [ 0.55, 1 ], "showticklabels": false, "title": { "font": { "size": 16 } } }, "xaxis4": { "anchor": "y4", "automargin": true, "autorange": true, "domain": [ 0.55, 1 ], "range": [ -0.6164347674215521, 9.616434767421552 ], "title": { "font": { "size": 16 }, "text": "Trial" }, "type": "linear" }, "yaxis": { "anchor": "x2", "automargin": true, "autorange": true, "domain": [ 0.31, 1 ], "range": [ -1.6912368304828413, 0.730886818235343 ], "title": { "font": { "size": 16 }, "text": "Score" }, "type": "linear" }, "yaxis2": { "anchor": "x2", "automargin": true, "autorange": true, "domain": [ 0, 0.29 ], "range": [ -2.341192660798647, 2.3708375798799204 ], "title": { "font": { "size": 16 }, "text": "d" }, "type": "linear" }, "yaxis3": { "anchor": "x4", "automargin": true, "autorange": true, "domain": [ 0.31, 1 ], "range": [ -5.227049478255329, -1.8780860730405433 ], "title": { "font": { "size": 16 }, "text": "Score" }, "type": "linear" }, "yaxis4": { "anchor": "x4", "automargin": true, "autorange": true, "domain": [ 0, 0.29 ], "range": [ -2.9844137736064407, 3.1476052717286582 ], "title": { "font": { "size": 16 }, "text": "d" }, "type": "linear" } } }, "text/html": [ "
" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "# Some plots allow us to choose the metric we want to show\n", "with atom.canvas():\n", " atom.plot_trials(metric=\"r2\", title=\"Hyperparameter tuning performance for R2\")\n", " atom.plot_trials(metric=\"rmse\", title=\"Hyperparameter tuning performance for RMSE\")" ] }, { "cell_type": "code", "execution_count": 9, "metadata": {}, "outputs": [ { "data": { "application/vnd.plotly.v1+json": { "config": { "plotlyServerURL": "https://plotly.com" }, "data": [ { "hovertemplate": "%{x}", "legendgroup": "r2", "marker": { "color": "rgba(0, 98, 98, 0.2)", "line": { "color": "rgb(0, 98, 98)", "width": 2 } }, "name": "r2", "orientation": "h", "showlegend": true, "type": "bar", "x": [ 0.4552, 0.5479 ], "xaxis": "x", "y": [ "lSVM", "hGBM" ], "yaxis": "y" } ], "layout": { "bargroupgap": 0.05, "font": { "size": 12 }, "height": 500, "hoverlabel": { "font": { "size": 16 } }, "legend": { "bgcolor": "rgba(255, 255, 255, 0.5)", "font": { "size": 16 }, "groupclick": "toggleitem", "traceorder": "grouped", "x": 0.99, "xanchor": "right", "y": 0.01, "yanchor": "bottom" }, "margin": { "b": 50, "l": 50, "pad": 0, "r": 0, "t": 35 }, "showlegend": true, "template": { "data": { "bar": [ { "error_x": { "color": "#2a3f5f" }, "error_y": { "color": "#2a3f5f" }, "marker": { "line": { "color": "#E5ECF6", "width": 0.5 }, "pattern": { "fillmode": "overlay", "size": 10, "solidity": 0.2 } }, "type": "bar" } ], "barpolar": [ { "marker": { "line": { "color": "#E5ECF6", "width": 0.5 }, "pattern": { "fillmode": "overlay", "size": 10, "solidity": 0.2 } }, "type": "barpolar" } ], "carpet": [ { "aaxis": { "endlinecolor": "#2a3f5f", "gridcolor": "white", "linecolor": "white", "minorgridcolor": "white", "startlinecolor": "#2a3f5f" }, "baxis": { "endlinecolor": "#2a3f5f", "gridcolor": "white", "linecolor": "white", "minorgridcolor": "white", "startlinecolor": "#2a3f5f" }, "type": "carpet" } ], "choropleth": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "type": "choropleth" } ], "contour": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "colorscale": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ], "type": "contour" } ], "contourcarpet": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "type": "contourcarpet" } ], "heatmap": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "colorscale": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ], "type": "heatmap" } ], "heatmapgl": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "colorscale": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ], "type": "heatmapgl" } ], "histogram": [ { "marker": { "pattern": { "fillmode": "overlay", "size": 10, "solidity": 0.2 } }, "type": "histogram" } ], "histogram2d": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "colorscale": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ], "type": "histogram2d" } ], "histogram2dcontour": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "colorscale": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ], "type": "histogram2dcontour" } ], "mesh3d": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "type": "mesh3d" } ], "parcoords": [ { "line": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "parcoords" } ], "pie": [ { "automargin": true, "type": "pie" } ], "scatter": [ { "fillpattern": { "fillmode": "overlay", "size": 10, "solidity": 0.2 }, "type": "scatter" } ], "scatter3d": [ { "line": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scatter3d" } ], "scattercarpet": [ { "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scattercarpet" } ], "scattergeo": [ { "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scattergeo" } ], "scattergl": [ { "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scattergl" } ], "scattermapbox": [ { "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scattermapbox" } ], "scatterpolar": [ { "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scatterpolar" } ], "scatterpolargl": [ { "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scatterpolargl" } ], "scatterternary": [ { "marker": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "type": "scatterternary" } ], "surface": [ { "colorbar": { "outlinewidth": 0, "ticks": "" }, "colorscale": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ], "type": "surface" } ], "table": [ { "cells": { "fill": { "color": "#EBF0F8" }, "line": { "color": "white" } }, "header": { "fill": { "color": "#C8D4E3" }, "line": { "color": "white" } }, "type": "table" } ] }, "layout": { "annotationdefaults": { "arrowcolor": "#2a3f5f", "arrowhead": 0, "arrowwidth": 1 }, "autotypenumbers": "strict", "coloraxis": { "colorbar": { "outlinewidth": 0, "ticks": "" } }, "colorscale": { "diverging": [ [ 0, "#8e0152" ], [ 0.1, "#c51b7d" ], [ 0.2, "#de77ae" ], [ 0.3, "#f1b6da" ], [ 0.4, "#fde0ef" ], [ 0.5, "#f7f7f7" ], [ 0.6, "#e6f5d0" ], [ 0.7, "#b8e186" ], [ 0.8, "#7fbc41" ], [ 0.9, "#4d9221" ], [ 1, "#276419" ] ], "sequential": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ], "sequentialminus": [ [ 0, "#0d0887" ], [ 0.1111111111111111, "#46039f" ], [ 0.2222222222222222, "#7201a8" ], [ 0.3333333333333333, "#9c179e" ], [ 0.4444444444444444, "#bd3786" ], [ 0.5555555555555556, "#d8576b" ], [ 0.6666666666666666, "#ed7953" ], [ 0.7777777777777778, "#fb9f3a" ], [ 0.8888888888888888, "#fdca26" ], [ 1, "#f0f921" ] ] }, "colorway": [ "#636efa", "#EF553B", "#00cc96", "#ab63fa", "#FFA15A", "#19d3f3", "#FF6692", "#B6E880", "#FF97FF", "#FECB52" ], "font": { "color": "#2a3f5f" }, "geo": { "bgcolor": "white", "lakecolor": "white", "landcolor": "#E5ECF6", "showlakes": true, "showland": true, "subunitcolor": "white" }, "hoverlabel": { "align": "left" }, "hovermode": "closest", "mapbox": { "style": "light" }, "paper_bgcolor": "white", "plot_bgcolor": "#E5ECF6", "polar": { "angularaxis": { "gridcolor": "white", "linecolor": "white", "ticks": "" }, "bgcolor": "#E5ECF6", "radialaxis": { "gridcolor": "white", "linecolor": "white", "ticks": "" } }, "scene": { "xaxis": { "backgroundcolor": "#E5ECF6", "gridcolor": "white", "gridwidth": 2, "linecolor": "white", "showbackground": true, "ticks": "", "zerolinecolor": "white" }, "yaxis": { "backgroundcolor": "#E5ECF6", "gridcolor": "white", "gridwidth": 2, "linecolor": "white", "showbackground": true, "ticks": "", "zerolinecolor": "white" }, "zaxis": { "backgroundcolor": "#E5ECF6", "gridcolor": "white", "gridwidth": 2, "linecolor": "white", "showbackground": true, "ticks": "", "zerolinecolor": "white" } }, "shapedefaults": { "line": { "color": "#2a3f5f" } }, "ternary": { "aaxis": { "gridcolor": "white", "linecolor": "white", "ticks": "" }, "baxis": { "gridcolor": "white", "linecolor": "white", "ticks": "" }, "bgcolor": "#E5ECF6", "caxis": { "gridcolor": "white", "linecolor": "white", "ticks": "" } }, "title": { "x": 0.05 }, "xaxis": { "automargin": true, "gridcolor": "white", "linecolor": "white", "ticks": "", "title": { "standoff": 15 }, "zerolinecolor": "white", "zerolinewidth": 2 }, "yaxis": { "automargin": true, "gridcolor": "white", "linecolor": "white", "ticks": "", "title": { "standoff": 15 }, "zerolinecolor": "white", "zerolinewidth": 2 } } }, "title": { "font": { "size": 24 }, "pad": { "b": 15, "t": 15 }, "x": 0.5, "xanchor": "center", "xref": "paper", "y": 1, "yanchor": "top" }, "width": 900, "xaxis": { "anchor": "y", "automargin": true, "autorange": true, "domain": [ 0, 1 ], "range": [ 0, 0.5767368421052632 ], "title": { "font": { "size": 16 }, "text": "score" }, "type": "linear" }, "yaxis": { "anchor": "x", "automargin": true, "autorange": true, "categoryorder": "total ascending", "domain": [ 0, 1 ], "range": [ -0.5, 1.5 ], "title": { "font": { "size": 16 } }, "type": "category" } } }, "text/html": [ "
" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "atom.plot_results(metric=\"r2\")" ] } ], "metadata": { "kernelspec": { "display_name": "Python 3 (ipykernel)", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.11.2" }, "toc": { "base_numbering": 1, "nav_menu": {}, "number_sections": true, "sideBar": true, "skip_h1_title": false, "title_cell": "Table of Contents", "title_sidebar": "Contents", "toc_cell": false, "toc_position": {}, "toc_section_display": true, "toc_window_display": false } }, "nbformat": 4, "nbformat_minor": 4 }