diff --git a/notebooks/rfpimp-collinear.ipynb b/notebooks/rfpimp-collinear.ipynb index 1a0688e..2e5fa52 100644 --- a/notebooks/rfpimp-collinear.ipynb +++ b/notebooks/rfpimp-collinear.ipynb @@ -2,7 +2,7 @@ "cells": [ { "cell_type": "code", - "execution_count": 2, + "execution_count": 1, "metadata": {}, "outputs": [], "source": [ @@ -20,7 +20,7 @@ }, { "cell_type": "code", - "execution_count": 3, + "execution_count": 2, "metadata": {}, "outputs": [], "source": [ @@ -34,7 +34,7 @@ }, { "cell_type": "code", - "execution_count": 4, + "execution_count": 3, "metadata": {}, "outputs": [], "source": [ @@ -56,7 +56,7 @@ }, { "cell_type": "code", - "execution_count": 5, + "execution_count": 4, "metadata": {}, "outputs": [ { @@ -90,8 +90,8 @@ "z\n", "\" style=\"fill:#ffffff;\"/>\n", " \n", - " \n", - " \n", + " \n", " \n", " \n", @@ -100,10 +100,10 @@ " \n", " \n", + "\" id=\"m184ff3767c\" style=\"stroke:#000000;stroke-width:0.8;\"/>\n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -320,7 +320,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -392,7 +392,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -452,7 +452,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -527,7 +527,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -593,7 +593,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -636,7 +636,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -668,7 +668,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -694,7 +694,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -720,10 +720,10 @@ " \n", " \n", + "\" id=\"m41d110b8a8\" style=\"stroke:#000000;stroke-width:0.8;\"/>\n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -744,7 +744,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -764,7 +764,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -784,7 +784,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -805,7 +805,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -822,7 +822,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -848,7 +848,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -880,7 +880,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -906,7 +906,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -1273,13 +1273,13 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -1423,13 +1423,13 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -1565,12 +1565,12 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -1622,7 +1622,7 @@ " \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", + "\" id=\"mbf620663c2\" style=\"stroke:#000000;stroke-width:0.8;\"/>\n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -1810,7 +1810,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -1825,7 +1825,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -1853,20 +1853,20 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", "\n" ], "text/plain": [ - "" + "" ] }, - "execution_count": 5, + "execution_count": 4, "metadata": {}, "output_type": "execute_result" } @@ -1880,7 +1880,7 @@ }, { "cell_type": "code", - "execution_count": 6, + "execution_count": 5, "metadata": { "scrolled": true }, @@ -1897,7 +1897,7 @@ " warm_start=False)" ] }, - "execution_count": 6, + "execution_count": 5, "metadata": {}, "output_type": "execute_result" } @@ -1922,7 +1922,7 @@ }, { "cell_type": "code", - "execution_count": 7, + "execution_count": 6, "metadata": {}, "outputs": [ { @@ -1957,66 +1957,66 @@ "\" style=\"fill:#ffffff;\"/>\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", @@ -2026,10 +2026,10 @@ " \n", " \n", + "\" id=\"m3ae3e61afa\" style=\"stroke:#000000;stroke-width:0.8;\"/>\n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -2063,7 +2063,7 @@ "z\n", "\" id=\"DejaVuSans-46\"/>\n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -2074,57 +2074,49 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", - " \n", + "\" id=\"DejaVuSans-53\"/>\n", " \n", - " \n", + " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -2143,30 +2135,6 @@ "L 12.40625 0 \n", "z\n", "\" id=\"DejaVuSans-49\"/>\n", - " \n", " \n", " \n", " \n", @@ -2183,14 +2151,14 @@ " \n", " \n", + "\" id=\"me377cd4e6c\" style=\"stroke:#000000;stroke-width:0.8;\"/>\n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", - " \n", - " \n", - " \n", - " \n", + "\" id=\"DejaVuSans-97\"/>\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", + "M 30.609375 56 \n", + "Q 42.328125 56 49.015625 48.375 \n", + "Q 55.71875 40.765625 55.71875 27.296875 \n", + "Q 55.71875 13.875 49.015625 6.21875 \n", + "Q 42.328125 -1.421875 30.609375 -1.421875 \n", + "Q 18.84375 -1.421875 12.171875 6.21875 \n", + "Q 5.515625 13.875 5.515625 27.296875 \n", + "Q 5.515625 40.765625 12.171875 48.375 \n", + "Q 18.84375 56 30.609375 56 \n", + "z\n", + "\" id=\"DejaVuSans-111\"/>\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", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", " \n", " \n", " \n", @@ -2659,28 +2584,75 @@ " \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", @@ -2737,15 +2688,15 @@ " \n", " \n", " \n", - " \n", - " \n", + " \n", + " \n", " \n", - " \n", + " \n", " \n", " \n", - " \n", + " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -2757,6 +2708,27 @@ " \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" ], "text/plain": [ - "" + "" ] }, - "execution_count": 7, + "execution_count": 6, "metadata": {}, "output_type": "execute_result" } @@ -2812,7 +2784,7 @@ }, { "cell_type": "code", - "execution_count": 8, + "execution_count": 7, "metadata": {}, "outputs": [ { @@ -2822,7 +2794,7 @@ "\n", "\n", - "\n", + "\n", " \n", " \n", " \n", " \n", - " \n", " \n", " \n", - " \n", " \n", " \n", - " \n", " \n", " \n", - " \n", " \n", " \n", - " \n", " \n", @@ -2876,10 +2848,10 @@ " \n", " \n", + "\" id=\"m29de094254\" style=\"stroke:#000000;stroke-width:0.8;\"/>\n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -2913,7 +2885,7 @@ "z\n", "\" id=\"DejaVuSans-46\"/>\n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -2924,7 +2896,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -2940,7 +2912,7 @@ "z\n", "\" id=\"DejaVuSans-55\"/>\n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -2951,7 +2923,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -2995,7 +2967,7 @@ "z\n", "\" id=\"DejaVuSans-53\"/>\n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -3010,57 +2982,73 @@ " \n", " \n", + "\" id=\"m0702fa039e\" style=\"stroke:#000000;stroke-width:0.8;\"/>\n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", - " \n", + " \n", - " \n", + "\" id=\"DejaVuSans-97\"/>\n", + " \n", " \n", - " \n", + "M 31.109375 56 \n", + "z\n", + "\" id=\"DejaVuSans-117\"/>\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", + "\" id=\"DejaVuSans-114\"/>\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", @@ -3413,124 +3382,129 @@ " \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", + "\" id=\"DejaVuSans-109\"/>\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", + "\" id=\"DejaVuSans-104\"/>\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", @@ -3541,39 +3515,39 @@ " \n", " \n", " \n", - " \n", " \n", " \n", - " \n", " \n", " \n", - " \n", " \n", " \n", - " \n", " \n", " \n", " \n", " \n", - " \n", - " \n", + " \n", + " \n", " \n", " \n", "\n" ], "text/plain": [ - "" + "" ] }, - "execution_count": 8, + "execution_count": 7, "metadata": {}, "output_type": "execute_result" } @@ -3587,7 +3561,7 @@ }, { "cell_type": "code", - "execution_count": 9, + "execution_count": 8, "metadata": {}, "outputs": [ { @@ -3622,17 +3596,17 @@ "\" style=\"fill:#ffffff;\"/>\n", " \n", " \n", - " \n", " \n", " \n", - " \n", @@ -3643,10 +3617,10 @@ " \n", " \n", + "\" id=\"m4ee80be172\" style=\"stroke:#000000;stroke-width:0.8;\"/>\n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -3691,34 +3665,55 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", - " \n", + "M 30.609375 32.421875 \n", + "Q 37.25 32.421875 41.125 36.953125 \n", + "Q 45.015625 41.5 45.015625 49.421875 \n", + "Q 45.015625 57.28125 41.125 61.84375 \n", + "Q 37.25 66.40625 30.609375 66.40625 \n", + "Q 23.96875 66.40625 20.09375 61.84375 \n", + "Q 16.21875 57.28125 16.21875 49.421875 \n", + "Q 16.21875 41.5 20.09375 36.953125 \n", + "Q 23.96875 32.421875 30.609375 32.421875 \n", + "z\n", + "\" id=\"DejaVuSans-57\"/>\n", " \n", - " \n", + " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -3760,32 +3755,144 @@ "Q 22.75 39.890625 18.8125 39.015625 \n", "Q 14.890625 38.140625 10.796875 36.28125 \n", "z\n", - "\" id=\"DejaVuSans-53\"/>\n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", - " \n", + "\" id=\"DejaVuSans-53\"/>\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", @@ -4117,7 +4131,26 @@ " \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", @@ -4163,48 +4184,24 @@ " \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", @@ -4321,17 +4318,17 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", "\n" ], "text/plain": [ - "" + "" ] }, - "execution_count": 9, + "execution_count": 8, "metadata": {}, "output_type": "execute_result" } @@ -4352,7 +4349,7 @@ }, { "cell_type": "code", - "execution_count": 10, + "execution_count": 9, "metadata": {}, "outputs": [ { @@ -4387,41 +4384,41 @@ "\" style=\"fill:#ffffff;\"/>\n", " \n", " \n", - " \n", " \n", " \n", - " \n", " \n", " \n", - " \n", " \n", " \n", - " \n", " \n", " \n", - " \n", @@ -4432,10 +4429,10 @@ " \n", " \n", + "\" id=\"maff7258ef3\" style=\"stroke:#000000;stroke-width:0.8;\"/>\n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -4480,34 +4477,64 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", - " \n", + "M 21.921875 38.8125 \n", + "Q 15.578125 40.375 12.03125 44.71875 \n", + "Q 8.5 49.078125 8.5 55.328125 \n", + "Q 8.5 64.0625 14.71875 69.140625 \n", + "Q 20.953125 74.21875 31.78125 74.21875 \n", + "Q 42.671875 74.21875 48.875 69.140625 \n", + "Q 55.078125 64.0625 55.078125 55.328125 \n", + "Q 55.078125 49.078125 51.53125 44.71875 \n", + "Q 48 40.375 41.703125 38.8125 \n", + "Q 48.828125 37.15625 52.796875 32.3125 \n", + "Q 56.78125 27.484375 56.78125 20.515625 \n", + "Q 56.78125 9.90625 50.3125 4.234375 \n", + "Q 43.84375 -1.421875 31.78125 -1.421875 \n", + "Q 19.734375 -1.421875 13.25 4.234375 \n", + "Q 6.78125 9.90625 6.78125 20.515625 \n", + "Q 6.78125 27.484375 10.78125 32.3125 \n", + "Q 14.796875 37.15625 21.921875 38.8125 \n", + "z\n", + "M 18.3125 54.390625 \n", + "Q 18.3125 48.734375 21.84375 45.5625 \n", + "Q 25.390625 42.390625 31.78125 42.390625 \n", + "Q 38.140625 42.390625 41.71875 45.5625 \n", + "Q 45.3125 48.734375 45.3125 54.390625 \n", + "Q 45.3125 60.0625 41.71875 63.234375 \n", + "Q 38.140625 66.40625 31.78125 66.40625 \n", + "Q 25.390625 66.40625 21.84375 63.234375 \n", + "Q 18.3125 60.0625 18.3125 54.390625 \n", + "z\n", + "\" id=\"DejaVuSans-56\"/>\n", " \n", - " \n", + " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -4566,10 +4593,10 @@ " \n", " \n", + "\" id=\"m3964976c59\" style=\"stroke:#000000;stroke-width:0.8;\"/>\n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -4818,7 +4845,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -5049,7 +5076,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -5086,7 +5113,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -5107,7 +5134,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -5148,17 +5175,17 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", "\n" ], "text/plain": [ - "" + "" ] }, - "execution_count": 10, + "execution_count": 9, "metadata": {}, "output_type": "execute_result" } @@ -5183,7 +5210,7 @@ }, { "cell_type": "code", - "execution_count": 11, + "execution_count": 10, "metadata": {}, "outputs": [ { @@ -5218,33 +5245,33 @@ "\" style=\"fill:#ffffff;\"/>\n", " \n", " \n", - " \n", " \n", " \n", - " \n", " \n", " \n", - " \n", " \n", " \n", - " \n", @@ -5255,10 +5282,10 @@ " \n", " \n", + "\" id=\"m7219bffecc\" style=\"stroke:#000000;stroke-width:0.8;\"/>\n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -5303,7 +5330,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -5319,7 +5346,7 @@ "z\n", "\" id=\"DejaVuSans-55\"/>\n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -5330,7 +5357,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -5389,10 +5416,10 @@ " \n", " \n", + "\" id=\"ma23abbe0d6\" style=\"stroke:#000000;stroke-width:0.8;\"/>\n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -5641,7 +5668,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -5892,7 +5919,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -5937,7 +5964,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -5979,17 +6006,17 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", "\n" ], "text/plain": [ - "" + "" ] }, - "execution_count": 11, + "execution_count": 10, "metadata": {}, "output_type": "execute_result" } @@ -6006,7 +6033,7 @@ }, { "cell_type": "code", - "execution_count": 13, + "execution_count": 11, "metadata": {}, "outputs": [ { @@ -6041,66 +6068,66 @@ "\" style=\"fill:#ffffff;\"/>\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", @@ -6110,10 +6137,10 @@ " \n", " \n", + "\" id=\"m6f8cddc323\" style=\"stroke:#000000;stroke-width:0.8;\"/>\n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -6147,7 +6174,7 @@ "z\n", "\" id=\"DejaVuSans-46\"/>\n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -6158,7 +6185,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -6182,7 +6209,7 @@ "z\n", "\" id=\"DejaVuSans-52\"/>\n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -6193,7 +6220,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -6252,10 +6279,10 @@ " \n", " \n", + "\" id=\"m44e3b579ab\" style=\"stroke:#000000;stroke-width:0.8;\"/>\n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -6459,7 +6486,7 @@ " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -6513,74 +6540,12 @@ " \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", @@ -6663,13 +6709,13 @@ " \n", " \n", " \n", - " \n", - " \n", + " \n", + " \n", " \n", - " \n", + " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -6727,31 +6773,14 @@ " \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", @@ -6786,15 +6837,15 @@ " \n", " \n", " \n", - " \n", - " \n", + " \n", + " \n", " \n", - " \n", + " \n", " \n", " \n", - " \n", + " \n", " \n", - " \n", + " \n", " \n", " \n", " \n", @@ -6806,28 +6857,6 @@ " \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" ], "text/plain": [ - "" + "" ] }, - "execution_count": 13, + "execution_count": 11, "metadata": {}, "output_type": "execute_result" } diff --git a/src/rfpimp.py b/src/rfpimp.py index 6a2d2cd..387571d 100644 --- a/src/rfpimp.py +++ b/src/rfpimp.py @@ -10,9 +10,10 @@ import pandas as pd import numpy as np import matplotlib.pyplot as plt +import sklearn from sklearn.ensemble import RandomForestClassifier from sklearn.ensemble import RandomForestRegressor -from sklearn.ensemble.forest import _generate_unsampled_indices, _get_n_samples_bootstrap +from sklearn.ensemble.forest import _generate_unsampled_indices from sklearn.ensemble import forest from sklearn.model_selection import cross_val_score from sklearn.base import clone @@ -415,13 +416,14 @@ def _get_unsampled_indices(tree, n_samples): """ An interface to get unsampled indices regardless of sklearn version. """ - try: # Version 0.21 or older uses only two arguments. - return _generate_unsampled_indices(tree.random_state, n_samples) - except TypeError: # Version 0.22 or newer uses only two arguments. + if sklearn.__version__.startswith("0.22"): + # Version 0.22 or newer uses 3 arguments. + from sklearn.ensemble.forest import _get_n_samples_bootstrap n_samples_bootstrap = _get_n_samples_bootstrap(n_samples, n_samples) - return _generate_unsampled_indices( - tree.random_state, n_samples, n_samples_bootstrap - ) + return _generate_unsampled_indices(tree.random_state, n_samples, n_samples_bootstrap) + else: + # Version 0.21 or older uses only two arguments. + return _generate_unsampled_indices(tree.random_state, n_samples) def oob_classifier_accuracy(rf, X_train, y_train):