{ "cells": [ { "cell_type": "markdown", "metadata": {}, "source": [ "# Predict Waist Circumference with Diffusion Weighted Imaging\n", "\n", "This notebook using diffusion weighted imaging data, and subjects waist circumference in cm from the ABCD Study.\n", "We will use as input feature derived Restriction spectrum imaging (RSI) from diffusion weighted images. This notebook\n", "covers data loading as well as evaluation across a large number of different ML Pipelines. This notebook may be useful\n", "for people looking for more examples on what different Pipelines to try." ] }, { "cell_type": "code", "execution_count": 1, "metadata": {}, "outputs": [], "source": [ "import BPt as bp\n", "import pandas as pd\n", "import os\n", "\n", "from warnings import simplefilter\n", "from sklearn.exceptions import ConvergenceWarning\n", "simplefilter(\"ignore\", category=ConvergenceWarning)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "## Load the data needed\n", "\n", "Data is loaded from a large csv file with all of the features from release 2 of the ABCD study." ] }, { "cell_type": "code", "execution_count": 2, "metadata": {}, "outputs": [], "source": [ "def load_from_rds(names, eventname='baseline_year_1_arm_1'):\n", " \n", " data = pd.read_csv('data/nda_rds_201.csv',\n", " usecols=['src_subject_id', 'eventname'] + names,\n", " na_values=['777', 999, '999', 777])\n", " \n", " data = data.loc[data[data['eventname'] == eventname].index]\n", " data = data.set_index('src_subject_id')\n", " data = data.drop('eventname', axis=1)\n", " \n", " # Obsificate subject ID for public example\n", " data.index = list(range(len(data)))\n", " \n", " # Return as pandas DataFrame cast to BPt Dataset\n", " return bp.Dataset(data)" ] }, { "cell_type": "code", "execution_count": 3, "metadata": {}, "outputs": [ { "data": { "text/plain": [ "['subjectid',\n", " 'src_subject_id',\n", " 'eventname',\n", " 'anthro_1_height_in',\n", " 'anthro_2_height_in',\n", " 'anthro_3_height_in',\n", " 'anthro_height_calc',\n", " 'anthro_weight_cast',\n", " 'anthro_weight_a_location',\n", " 'anthro_weight1_lb']" ] }, "execution_count": 3, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# This way we can look at all column available\n", "all_cols = list(pd.read_csv('data/nda_rds_201.csv', nrows=0))\n", "all_cols[:10]" ] }, { "cell_type": "code", "execution_count": 4, "metadata": {}, "outputs": [ { "data": { "text/plain": [ "294" ] }, "execution_count": 4, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# The target variable\n", "target_cols = ['anthro_waist_cm']\n", "\n", "# non input feature - i.e., those that inform \n", "non_input_cols = ['sex', 'rel_family_id']\n", "\n", "# We will use the fiber at dti measures\n", "dti_cols = [c for c in all_cols if '_fiber.at' in c and 'rsi.' in c]\n", "len(dti_cols)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "Now we can use the helper function defined at the start to load these features in as a Dataset" ] }, { "cell_type": "code", "execution_count": 5, "metadata": {}, "outputs": [ { "data": { "text/plain": [ "(11875, 297)" ] }, "execution_count": 5, "metadata": {}, "output_type": "execute_result" } ], "source": [ "data = load_from_rds(target_cols + non_input_cols + dti_cols)\n", "data.shape" ] }, { "cell_type": "code", "execution_count": 6, "metadata": {}, "outputs": [], "source": [ "# This is optional, but will print out some extra verbosity when using the dataset operations\n", "data.verbose = 1" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "The first step we will do is tell the dataset what roles the different columns are. See: https://sahahn.github.io/BPt/user_guide/role.html" ] }, { "cell_type": "code", "execution_count": 7, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Dropped 2 Rows\n", "Dropped 6 Rows\n" ] }, { "data": { "text/html": [ "
\n", " | dmri_rsi.n0_fiber.at_allfib.lh | \n", "dmri_rsi.n0_fiber.at_allfib.rh | \n", "dmri_rsi.n0_fiber.at_allfibers | \n", "dmri_rsi.n0_fiber.at_allfibnocc.lh | \n", "dmri_rsi.n0_fiber.at_allfibnocc.rh | \n", "dmri_rsi.n0_fiber.at_atr.lh | \n", "dmri_rsi.n0_fiber.at_atr.rh | \n", "dmri_rsi.n0_fiber.at_cc | \n", "dmri_rsi.n0_fiber.at_cgc.lh | \n", "dmri_rsi.n0_fiber.at_cgc.rh | \n", "... | \n", "dmri_rsi.vol_fiber.at_scs.lh | \n", "dmri_rsi.vol_fiber.at_scs.rh | \n", "dmri_rsi.vol_fiber.at_sifc.lh | \n", "dmri_rsi.vol_fiber.at_sifc.rh | \n", "dmri_rsi.vol_fiber.at_slf.lh | \n", "dmri_rsi.vol_fiber.at_slf.rh | \n", "dmri_rsi.vol_fiber.at_tslf.lh | \n", "dmri_rsi.vol_fiber.at_tslf.rh | \n", "dmri_rsi.vol_fiber.at_unc.lh | \n", "dmri_rsi.vol_fiber.at_unc.rh | \n", "
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
0 | \n", "0.327623 | \n", "0.323420 | \n", "0.325957 | \n", "0.340559 | \n", "0.332364 | \n", "0.347837 | \n", "0.336072 | \n", "0.306803 | \n", "0.311347 | \n", "0.304854 | \n", "... | \n", "23672.0 | \n", "13056.0 | \n", "9648.0 | \n", "9528.0 | \n", "10152.0 | \n", "11504.0 | \n", "8384.0 | \n", "8024.0 | \n", "4968.0 | \n", "7176.0 | \n", "
1 | \n", "NaN | \n", "NaN | \n", "NaN | \n", "NaN | \n", "NaN | \n", "NaN | \n", "NaN | \n", "NaN | \n", "NaN | \n", "NaN | \n", "... | \n", "NaN | \n", "NaN | \n", "NaN | \n", "NaN | \n", "NaN | \n", "NaN | \n", "NaN | \n", "NaN | \n", "NaN | \n", "NaN | \n", "
2 | \n", "0.325374 | \n", "0.311465 | \n", "0.319027 | \n", "0.341213 | \n", "0.326334 | \n", "0.346651 | \n", "0.335362 | \n", "0.288124 | \n", "0.326416 | \n", "0.300990 | \n", "... | \n", "33112.0 | \n", "19256.0 | \n", "11928.0 | \n", "8688.0 | \n", "13144.0 | \n", "15344.0 | \n", "10488.0 | \n", "10936.0 | \n", "6904.0 | \n", "9480.0 | \n", "
3 | \n", "0.305095 | \n", "0.304357 | \n", "0.305170 | \n", "0.315477 | \n", "0.312866 | \n", "0.313972 | \n", "0.316729 | \n", "0.288742 | \n", "0.289166 | \n", "0.290347 | \n", "... | \n", "28480.0 | \n", "16016.0 | \n", "13024.0 | \n", "11960.0 | \n", "13600.0 | \n", "14880.0 | \n", "11416.0 | \n", "10592.0 | \n", "6952.0 | \n", "8736.0 | \n", "
4 | \n", "0.316860 | \n", "0.315238 | \n", "0.316399 | \n", "0.328251 | \n", "0.327259 | \n", "0.333998 | \n", "0.318162 | \n", "0.294008 | \n", "0.297800 | \n", "0.299230 | \n", "... | \n", "29904.0 | \n", "17968.0 | \n", "12720.0 | \n", "11336.0 | \n", "13528.0 | \n", "15672.0 | \n", "11096.0 | \n", "11816.0 | \n", "5912.0 | \n", "7336.0 | \n", "
... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "
11870 | \n", "0.335741 | \n", "0.336048 | \n", "0.336372 | \n", "0.349806 | \n", "0.347732 | \n", "0.349966 | \n", "0.345692 | \n", "0.312056 | \n", "0.324054 | \n", "0.336676 | \n", "... | \n", "28328.0 | \n", "15400.0 | \n", "9656.0 | \n", "10080.0 | \n", "11312.0 | \n", "13496.0 | \n", "8728.0 | \n", "9176.0 | \n", "4960.0 | \n", "7392.0 | \n", "
11871 | \n", "0.320563 | \n", "0.317525 | \n", "0.319429 | \n", "0.327302 | \n", "0.322161 | \n", "0.333086 | \n", "0.315482 | \n", "0.308554 | \n", "0.299938 | \n", "0.298093 | \n", "... | \n", "23792.0 | \n", "13632.0 | \n", "9928.0 | \n", "8912.0 | \n", "9152.0 | \n", "12288.0 | \n", "7128.0 | \n", "8912.0 | \n", "5744.0 | \n", "7376.0 | \n", "
11872 | \n", "0.327051 | \n", "0.325386 | \n", "0.326522 | \n", "0.340918 | \n", "0.334854 | \n", "0.345435 | \n", "0.335610 | \n", "0.305720 | \n", "0.308630 | \n", "0.330612 | \n", "... | \n", "28640.0 | \n", "16384.0 | \n", "9496.0 | \n", "11216.0 | \n", "12168.0 | \n", "12312.0 | \n", "9520.0 | \n", "8952.0 | \n", "4568.0 | \n", "9056.0 | \n", "
11873 | \n", "0.323579 | \n", "0.319377 | \n", "0.321805 | \n", "0.334945 | \n", "0.329433 | \n", "0.332200 | \n", "0.334017 | \n", "0.304399 | \n", "0.303831 | \n", "0.307037 | \n", "... | \n", "26216.0 | \n", "14672.0 | \n", "9408.0 | \n", "8872.0 | \n", "10960.0 | \n", "12584.0 | \n", "8880.0 | \n", "9176.0 | \n", "3696.0 | \n", "6168.0 | \n", "
11874 | \n", "0.383537 | \n", "0.371483 | \n", "0.377822 | \n", "0.394413 | \n", "0.373486 | \n", "0.404684 | \n", "0.374435 | \n", "0.366270 | \n", "0.415342 | \n", "0.418280 | \n", "... | \n", "26544.0 | \n", "15624.0 | \n", "9904.0 | \n", "10360.0 | \n", "9904.0 | \n", "12216.0 | \n", "7712.0 | \n", "8000.0 | \n", "5208.0 | \n", "8816.0 | \n", "
11867 rows × 294 columns
\n", "\n", " | anthro_waist_cm | \n", "
---|---|
0 | \n", "31.00 | \n", "
1 | \n", "30.50 | \n", "
2 | \n", "26.75 | \n", "
3 | \n", "23.50 | \n", "
4 | \n", "30.00 | \n", "
... | \n", "... | \n", "
11870 | \n", "26.00 | \n", "
11871 | \n", "30.00 | \n", "
11872 | \n", "19.00 | \n", "
11873 | \n", "25.00 | \n", "
11874 | \n", "32.00 | \n", "
11867 rows × 1 columns
\n", "\n", " | rel_family_id | \n", "sex | \n", "
---|---|---|
0 | \n", "8780.0 | \n", "F | \n", "
1 | \n", "10207.0 | \n", "F | \n", "
2 | \n", "4720.0 | \n", "M | \n", "
3 | \n", "3804.0 | \n", "M | \n", "
4 | \n", "5358.0 | \n", "M | \n", "
... | \n", "... | \n", "... | \n", "
11870 | \n", "3791.0 | \n", "M | \n", "
11871 | \n", "2441.0 | \n", "F | \n", "
11872 | \n", "7036.0 | \n", "F | \n", "
11873 | \n", "6681.0 | \n", "F | \n", "
11874 | \n", "7588.0 | \n", "F | \n", "
11867 rows × 2 columns
\n", "\n", " | rel_family_id | \n", "sex | \n", "
---|---|---|
0 | \n", "7321 | \n", "0 | \n", "
1 | \n", "8634 | \n", "0 | \n", "
2 | \n", "3971 | \n", "1 | \n", "
3 | \n", "3139 | \n", "1 | \n", "
4 | \n", "4543 | \n", "1 | \n", "
... | \n", "... | \n", "... | \n", "
11870 | \n", "3128 | \n", "1 | \n", "
11871 | \n", "2111 | \n", "0 | \n", "
11872 | \n", "5907 | \n", "0 | \n", "
11873 | \n", "5594 | \n", "0 | \n", "
11874 | \n", "6238 | \n", "0 | \n", "
11867 rows × 2 columns
\n", "\n", " | dmri_rsi.n0_fiber.at_allfib.lh | \n", "dmri_rsi.n0_fiber.at_allfib.rh | \n", "dmri_rsi.n0_fiber.at_allfibers | \n", "dmri_rsi.n0_fiber.at_allfibnocc.lh | \n", "dmri_rsi.n0_fiber.at_allfibnocc.rh | \n", "dmri_rsi.n0_fiber.at_atr.lh | \n", "dmri_rsi.n0_fiber.at_atr.rh | \n", "dmri_rsi.n0_fiber.at_cc | \n", "dmri_rsi.n0_fiber.at_cgc.lh | \n", "dmri_rsi.n0_fiber.at_cgc.rh | \n", "... | \n", "dmri_rsi.vol_fiber.at_scs.lh | \n", "dmri_rsi.vol_fiber.at_scs.rh | \n", "dmri_rsi.vol_fiber.at_sifc.lh | \n", "dmri_rsi.vol_fiber.at_sifc.rh | \n", "dmri_rsi.vol_fiber.at_slf.lh | \n", "dmri_rsi.vol_fiber.at_slf.rh | \n", "dmri_rsi.vol_fiber.at_tslf.lh | \n", "dmri_rsi.vol_fiber.at_tslf.rh | \n", "dmri_rsi.vol_fiber.at_unc.lh | \n", "dmri_rsi.vol_fiber.at_unc.rh | \n", "
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
0 | \n", "0.327623 | \n", "0.323420 | \n", "0.325957 | \n", "0.340559 | \n", "0.332364 | \n", "0.347837 | \n", "0.336072 | \n", "0.306803 | \n", "0.311347 | \n", "0.304854 | \n", "... | \n", "23672.0 | \n", "13056.0 | \n", "9648.0 | \n", "9528.0 | \n", "10152.0 | \n", "11504.0 | \n", "8384.0 | \n", "8024.0 | \n", "4968.0 | \n", "7176.0 | \n", "
2 | \n", "0.325374 | \n", "0.311465 | \n", "0.319027 | \n", "0.341213 | \n", "0.326334 | \n", "0.346651 | \n", "0.335362 | \n", "0.288124 | \n", "0.326416 | \n", "0.300990 | \n", "... | \n", "33112.0 | \n", "19256.0 | \n", "11928.0 | \n", "8688.0 | \n", "13144.0 | \n", "15344.0 | \n", "10488.0 | \n", "10936.0 | \n", "6904.0 | \n", "9480.0 | \n", "
3 | \n", "0.305095 | \n", "0.304357 | \n", "0.305170 | \n", "0.315477 | \n", "0.312866 | \n", "0.313972 | \n", "0.316729 | \n", "0.288742 | \n", "0.289166 | \n", "0.290347 | \n", "... | \n", "28480.0 | \n", "16016.0 | \n", "13024.0 | \n", "11960.0 | \n", "13600.0 | \n", "14880.0 | \n", "11416.0 | \n", "10592.0 | \n", "6952.0 | \n", "8736.0 | \n", "
4 | \n", "0.316860 | \n", "0.315238 | \n", "0.316399 | \n", "0.328251 | \n", "0.327259 | \n", "0.333998 | \n", "0.318162 | \n", "0.294008 | \n", "0.297800 | \n", "0.299230 | \n", "... | \n", "29904.0 | \n", "17968.0 | \n", "12720.0 | \n", "11336.0 | \n", "13528.0 | \n", "15672.0 | \n", "11096.0 | \n", "11816.0 | \n", "5912.0 | \n", "7336.0 | \n", "
5 | \n", "0.323521 | \n", "0.326741 | \n", "0.325466 | \n", "0.336003 | \n", "0.335291 | \n", "0.326243 | \n", "0.337367 | \n", "0.305382 | \n", "0.311843 | \n", "0.315721 | \n", "... | \n", "23048.0 | \n", "12032.0 | \n", "9056.0 | \n", "9248.0 | \n", "9672.0 | \n", "11048.0 | \n", "7848.0 | \n", "7520.0 | \n", "5088.0 | \n", "7448.0 | \n", "
... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "
11870 | \n", "0.335741 | \n", "0.336048 | \n", "0.336372 | \n", "0.349806 | \n", "0.347732 | \n", "0.349966 | \n", "0.345692 | \n", "0.312056 | \n", "0.324054 | \n", "0.336676 | \n", "... | \n", "28328.0 | \n", "15400.0 | \n", "9656.0 | \n", "10080.0 | \n", "11312.0 | \n", "13496.0 | \n", "8728.0 | \n", "9176.0 | \n", "4960.0 | \n", "7392.0 | \n", "
11871 | \n", "0.320563 | \n", "0.317525 | \n", "0.319429 | \n", "0.327302 | \n", "0.322161 | \n", "0.333086 | \n", "0.315482 | \n", "0.308554 | \n", "0.299938 | \n", "0.298093 | \n", "... | \n", "23792.0 | \n", "13632.0 | \n", "9928.0 | \n", "8912.0 | \n", "9152.0 | \n", "12288.0 | \n", "7128.0 | \n", "8912.0 | \n", "5744.0 | \n", "7376.0 | \n", "
11872 | \n", "0.327051 | \n", "0.325386 | \n", "0.326522 | \n", "0.340918 | \n", "0.334854 | \n", "0.345435 | \n", "0.335610 | \n", "0.305720 | \n", "0.308630 | \n", "0.330612 | \n", "... | \n", "28640.0 | \n", "16384.0 | \n", "9496.0 | \n", "11216.0 | \n", "12168.0 | \n", "12312.0 | \n", "9520.0 | \n", "8952.0 | \n", "4568.0 | \n", "9056.0 | \n", "
11873 | \n", "0.323579 | \n", "0.319377 | \n", "0.321805 | \n", "0.334945 | \n", "0.329433 | \n", "0.332200 | \n", "0.334017 | \n", "0.304399 | \n", "0.303831 | \n", "0.307037 | \n", "... | \n", "26216.0 | \n", "14672.0 | \n", "9408.0 | \n", "8872.0 | \n", "10960.0 | \n", "12584.0 | \n", "8880.0 | \n", "9176.0 | \n", "3696.0 | \n", "6168.0 | \n", "
11874 | \n", "0.383537 | \n", "0.371483 | \n", "0.377822 | \n", "0.394413 | \n", "0.373486 | \n", "0.404684 | \n", "0.374435 | \n", "0.366270 | \n", "0.415342 | \n", "0.418280 | \n", "... | \n", "26544.0 | \n", "15624.0 | \n", "9904.0 | \n", "10360.0 | \n", "9904.0 | \n", "12216.0 | \n", "7712.0 | \n", "8000.0 | \n", "5208.0 | \n", "8816.0 | \n", "
10663 rows × 294 columns
\n", "8562 rows × 294 columns - Train Set
2101 rows × 294 columns - Test Set
\n", " | anthro_waist_cm | \n", "
---|---|
0 | \n", "31.00 | \n", "
2 | \n", "26.75 | \n", "
3 | \n", "23.50 | \n", "
4 | \n", "30.00 | \n", "
5 | \n", "28.00 | \n", "
... | \n", "... | \n", "
11870 | \n", "26.00 | \n", "
11871 | \n", "30.00 | \n", "
11872 | \n", "19.00 | \n", "
11873 | \n", "25.00 | \n", "
11874 | \n", "32.00 | \n", "
10663 rows × 1 columns
\n", "8562 rows × 1 columns - Train Set
2101 rows × 1 columns - Test Set
\n", " | rel_family_id | \n", "sex | \n", "
---|---|---|
0 | \n", "7321 | \n", "0 | \n", "
2 | \n", "3971 | \n", "1 | \n", "
3 | \n", "3139 | \n", "1 | \n", "
4 | \n", "4543 | \n", "1 | \n", "
5 | \n", "1933 | \n", "1 | \n", "
... | \n", "... | \n", "... | \n", "
11870 | \n", "3128 | \n", "1 | \n", "
11871 | \n", "2111 | \n", "0 | \n", "
11872 | \n", "5907 | \n", "0 | \n", "
11873 | \n", "5594 | \n", "0 | \n", "
11874 | \n", "6238 | \n", "0 | \n", "
10663 rows × 2 columns
\n", "8562 rows × 2 columns - Train Set
2101 rows × 2 columns - Test Set
\n", " | dmri_rsi.n0_fiber.at_allfib.lh | \n", "dmri_rsi.n0_fiber.at_allfib.rh | \n", "dmri_rsi.n0_fiber.at_allfibers | \n", "dmri_rsi.n0_fiber.at_allfibnocc.lh | \n", "dmri_rsi.n0_fiber.at_allfibnocc.rh | \n", "dmri_rsi.n0_fiber.at_atr.lh | \n", "dmri_rsi.n0_fiber.at_atr.rh | \n", "dmri_rsi.n0_fiber.at_cc | \n", "dmri_rsi.n0_fiber.at_cgc.lh | \n", "dmri_rsi.n0_fiber.at_cgc.rh | \n", "... | \n", "dmri_rsi.vol_fiber.at_scs.lh | \n", "dmri_rsi.vol_fiber.at_scs.rh | \n", "dmri_rsi.vol_fiber.at_sifc.lh | \n", "dmri_rsi.vol_fiber.at_sifc.rh | \n", "dmri_rsi.vol_fiber.at_slf.lh | \n", "dmri_rsi.vol_fiber.at_slf.rh | \n", "dmri_rsi.vol_fiber.at_tslf.lh | \n", "dmri_rsi.vol_fiber.at_tslf.rh | \n", "dmri_rsi.vol_fiber.at_unc.lh | \n", "dmri_rsi.vol_fiber.at_unc.rh | \n", "
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
0 | \n", "13.658717 | \n", "10.182917 | \n", "11.113703 | \n", "2.741103 | \n", "2.096137 | \n", "0.652331 | \n", "-19.689333 | \n", "18.133593 | \n", "6.170156 | \n", "18.584436 | \n", "... | \n", "0.000343 | \n", "-0.000589 | \n", "0.000551 | \n", "0.000348 | \n", "-0.000216 | \n", "0.000278 | \n", "1.902580e-04 | \n", "-0.000544 | \n", "0.000643 | \n", "0.000572 | \n", "
1 | \n", "13.386769 | \n", "-7.807120 | \n", "5.590872 | \n", "2.074826 | \n", "1.611921 | \n", "-5.200155 | \n", "-17.350840 | \n", "18.954327 | \n", "4.418044 | \n", "18.556810 | \n", "... | \n", "0.000263 | \n", "-0.000505 | \n", "0.000523 | \n", "0.000272 | \n", "-0.000371 | \n", "0.000474 | \n", "1.721382e-04 | \n", "-0.000582 | \n", "0.000530 | \n", "0.000525 | \n", "
2 | \n", "14.601005 | \n", "6.447403 | \n", "6.566593 | \n", "5.105803 | \n", "2.144557 | \n", "-0.249971 | \n", "-17.505516 | \n", "19.815592 | \n", "10.121972 | \n", "15.690188 | \n", "... | \n", "0.000319 | \n", "-0.000600 | \n", "0.000455 | \n", "0.000446 | \n", "-0.000303 | \n", "0.000618 | \n", "1.192093e-07 | \n", "-0.000615 | \n", "0.000378 | \n", "0.000662 | \n", "
3 | \n", "12.660464 | \n", "16.193197 | \n", "-0.193520 | \n", "2.184492 | \n", "3.968235 | \n", "-7.643031 | \n", "-16.103781 | \n", "15.539372 | \n", "9.867133 | \n", "14.523899 | \n", "... | \n", "0.000341 | \n", "-0.000441 | \n", "0.000587 | \n", "0.000453 | \n", "-0.000284 | \n", "0.000608 | \n", "1.287460e-04 | \n", "-0.000701 | \n", "0.000359 | \n", "0.000616 | \n", "
4 | \n", "16.212349 | \n", "11.518905 | \n", "7.905156 | \n", "6.085897 | \n", "1.294112 | \n", "-8.058172 | \n", "-11.786942 | \n", "13.227197 | \n", "10.812229 | \n", "12.374775 | \n", "... | \n", "0.000080 | \n", "-0.000515 | \n", "0.000614 | \n", "0.000411 | \n", "-0.000342 | \n", "0.000784 | \n", "2.737045e-04 | \n", "-0.000738 | \n", "0.000555 | \n", "0.000540 | \n", "
5 rows × 294 columns
\n", "\n", " | predict | \n", "y_true | \n", "
---|---|---|
28 | \n", "24.345831 | \n", "21.25 | \n", "
33 | \n", "24.806902 | \n", "24.50 | \n", "
36 | \n", "26.968584 | \n", "23.00 | \n", "
40 | \n", "27.805948 | \n", "30.80 | \n", "
47 | \n", "25.691679 | \n", "20.00 | \n", "
... | \n", "... | \n", "... | \n", "
11848 | \n", "28.006598 | \n", "25.50 | \n", "
11851 | \n", "23.925535 | \n", "24.00 | \n", "
11855 | \n", "30.389437 | \n", "38.80 | \n", "
11861 | \n", "25.687609 | \n", "26.00 | \n", "
11868 | \n", "29.838160 | \n", "35.00 | \n", "
1713 rows × 2 columns
\n", "\n", " | mean_scores_explained_variance | \n", "mean_scores_neg_mean_squared_error | \n", "std_scores_explained_variance | \n", "std_scores_neg_mean_squared_error | \n", "mean_timing_fit | \n", "mean_timing_score | \n", "
---|---|---|---|---|---|---|
pipeline | \n", "\n", " | \n", " | \n", " | \n", " | \n", " | \n", " |
sgd | \n", "0.226867 | \n", "-13.632093 | \n", "0.015547 | \n", "0.287249 | \n", "16.182405 | \n", "0.007756 | \n", "
ridge | \n", "0.244241 | \n", "-13.105543 | \n", "0.014348 | \n", "0.335205 | \n", "11.024628 | \n", "0.461483 | \n", "
elastic | \n", "0.214935 | \n", "-13.612086 | \n", "0.017457 | \n", "0.304521 | \n", "8.584299 | \n", "0.036844 | \n", "
lgbm | \n", "0.159819 | \n", "-14.628175 | \n", "0.008612 | \n", "0.317664 | \n", "30.417882 | \n", "0.063121 | \n", "