1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | import numpy as np import matplotlib.pyplot as plt import scipy.stats from sklearn.linear_model import LogisticRegression from sklearn.metrics import ( accuracy_score, balanced_accuracy_score, confusion_matrix ) import pandas as pd import seaborn as sns from sklearn.linear_model import LinearRegression, RidgeCV from sklearn.metrics import r2_score from sklearn.model_selection import ShuffleSplit, KFold |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | def generate_classification_results(n=1000, seed=None, noise_sd=1, signal_sd=10, p_pos=0.5, shuffle=False, pred=None, clf=None): rng = np.random.RandomState(seed) X = rng.normal(size=n) * signal_sd noise = noise_sd * rng.randn(n) thresh = scipy.stats.scoreatpercentile(X, 100 * (1 - p_pos)) y = ((X + noise) > thresh).astype('int') if shuffle: rng.shuffle(y) if pred is not None: y_pred = pred * np.ones(n) else: clf = LogisticRegression() clf.fit(X[:, None], y) y_pred = clf.predict(X[:, None]) results = { 'acc': accuracy_score(y, y_pred), 'bacc': balanced_accuracy_score(y, y_pred), 'confusion': confusion_matrix(y, y_pred), 'n': n, 'seed': seed, 'noise_sd': noise_sd, 'signal_sd': signal_sd, 'p_pos': p_pos, 'shuffle': shuffle, 'thresh': thresh, 'clf': clf, 'X': X, 'y': y, 'pred': y_pred, } return(results) |
1 2 | r = generate_classification_results(p_pos=0.9) r |
{'acc': 0.984, 'bacc': np.float64(0.9481118290742605), 'confusion': array([[ 93, 10], [ 6, 891]]), 'n': 1000, 'seed': None, 'noise_sd': 1, 'signal_sd': 10, 'p_pos': 0.9, 'shuffle': False, 'thresh': np.float64(-13.026289334267915), 'clf': LogisticRegression(), 'X': array([ 1.64182403e+01, 7.73439060e+00, 2.24248539e+00, -8.63066851e+00, 8.08734003e+00, 3.05031670e+00, -5.24783181e+00, -1.26985283e+01, 7.02741658e+00, -5.04099515e+00, -4.51292740e+00, 2.11733753e+01, 7.91278931e+00, -1.28838295e+01, -2.11326839e+01, -6.35590505e-01, -7.44629224e+00, -1.50205882e+01, -5.10118865e+00, 5.97476744e+00, 1.05675346e+01, 8.82422093e+00, -9.99421920e+00, 1.78710953e+01, 7.54784544e+00, -9.46454849e+00, 2.10661502e+00, 1.69243192e+01, 1.33725754e+01, -2.65811557e+00, -6.89942099e+00, 4.56516278e+00, 5.36624832e+00, -1.99801218e+00, -5.78889359e+00, 7.52852491e+00, -1.38892648e+01, 2.31451016e+00, 2.14269986e+00, 3.14156627e+00, 3.64628031e+00, 3.13339286e+00, -9.19555068e+00, 3.50681658e+00, 1.78900125e+00, -7.62436095e+00, 1.91105042e+00, -1.96914608e+01, 2.65339338e+00, -8.99807889e+00, 4.39679224e+00, 2.98363465e-01, 4.36490647e-01, -1.22517331e+01, 2.49818559e+00, 1.05609829e+01, -1.61199350e+00, -8.78655063e+00, -3.62751109e-01, -6.30486686e+00, -6.55623218e+00, 9.13852545e+00, -3.54492009e+00, -6.69395499e+00, 1.03572708e+00, 3.29768479e+00, -1.49825315e+01, -6.32321316e+00, -7.62370810e+00, -6.48895970e+00, -3.69362388e+00, -1.11737532e+01, -1.20850957e+01, 1.10429545e+01, -2.32439837e+01, -8.78555988e-02, 5.33577136e+00, 1.34301949e+00, -2.07453617e+00, -8.86308612e+00, 1.05551117e+01, 1.81219366e-01, 6.65255847e-01, 1.28734506e+00, -2.52616051e+00, -4.67388873e+00, -3.11685146e+00, -2.20452620e+00, -7.04574788e+00, 6.51193142e+00, 6.22291664e+00, 1.69545423e+01, 1.50267226e+01, -1.23496888e+01, -4.18022403e+00, -6.21144299e+00, 2.96953188e+00, 2.76768861e+00, 5.83285742e+00, -6.60230362e+00, 2.38212431e+00, -8.52558914e+00, -1.46343942e+01, 7.22614066e+00, 2.07625114e+01, 6.52645879e+00, 9.70397858e-03, 1.22505519e+01, 1.51175988e+01, 1.05256108e+00, 5.80415404e+00, 8.20465950e+00, -1.09780785e+01, 1.02651068e+01, -1.57728798e+01, 1.22322274e+01, -4.48691964e+00, 8.28831454e+00, -1.26527682e+01, -1.70314719e+01, -4.56979367e+00, 3.36759968e+00, -1.43782113e+01, -4.69820016e+00, -7.58383312e+00, 6.18271217e+00, 2.42078924e+01, -4.55935759e+00, -3.70315619e+00, 1.29309556e+00, 9.62816266e+00, 8.40026453e+00, -6.39934629e+00, -2.42885780e+01, 1.48434393e+01, -1.30669028e+01, -5.78598149e+00, -5.33971676e-02, -9.34195974e-01, -1.24292071e+01, -9.61328630e+00, 9.47001899e-01, 8.34116356e+00, 8.65824801e+00, 1.28922269e+00, -1.03818377e+01, -1.30217767e+01, -6.56708827e-01, -4.19929802e+00, 4.92283959e+00, 8.98451719e+00, 1.55046355e+01, -2.75781153e+01, 2.33474635e+00, 6.32197759e+00, 1.28563615e+01, -6.45290440e+00, -3.42474458e-01, -1.36012980e+00, -5.79145629e+00, 2.44182234e+01, 3.23469003e+00, -1.99237463e+01, -1.05342888e+00, -9.58588278e+00, -4.26816928e-01, -6.74501224e+00, -4.78747806e-01, 3.05755059e+00, 1.66292221e+01, -1.63424490e+00, 5.41252228e+00, -1.18903510e+01, 9.38959641e+00, 2.80925611e+00, 1.37291393e+00, -5.84519874e-01, -7.30336972e+00, -4.85723603e-01, -1.34160815e+00, -1.45983435e+01, -1.22040214e+01, -7.57846092e-01, -1.45623626e+01, -4.99445682e+00, -1.23398955e+00, -4.53285665e+00, -9.30056784e-01, -4.67155920e+00, 7.68464081e+00, -1.94471631e+01, -1.97629688e+01, -1.45483425e+01, 3.29865499e+00, 1.65723486e+00, 9.26982201e+00, 1.08963305e+01, 2.08242109e+01, 8.19315738e-01, 2.64974643e-02, 3.38661233e+00, 1.07683308e+00, 2.79009996e+00, 8.68033133e+00, -3.57874125e+00, -3.90450692e+00, 1.71548647e+00, 2.88052290e+00, 1.20138560e+01, 1.90818169e+01, -2.99106572e+00, -8.33209491e+00, -9.51825221e+00, -5.50318977e+00, -7.23905065e+00, 3.14791062e-01, 3.80503909e+00, -2.15112697e+00, -2.23392111e+01, -2.50626014e+00, -2.57191957e+00, 7.23503151e+00, -5.03231304e+00, -8.92030601e+00, -6.50431853e+00, -1.91500605e+00, 2.07385265e-01, -2.57743495e+01, -9.89153644e+00, 4.74615923e-01, -1.71812547e+01, 9.45231213e+00, 7.68634817e+00, 1.65723415e+01, 1.40461848e+01, -1.01074514e+01, -5.86829610e+00, -1.84656216e+01, -1.85396966e+01, 9.11069858e-01, -7.67113094e+00, -2.58634434e+01, -5.67436239e+00, -6.51314843e+00, -5.54978358e+00, -9.47094713e+00, -6.34636676e-01, 1.38817857e+00, 3.93110268e+00, -1.86255085e+01, 7.45523586e+00, -9.47202193e+00, -8.68205929e+00, 9.82781403e+00, -9.29555834e+00, -1.73566647e+00, -2.24575020e+01, 1.06474264e+01, 5.26833212e+00, -2.46911947e+00, 3.30997735e+00, -3.09713924e+00, -2.07604822e+00, -1.32362087e+01, 7.51746786e+00, 1.09485991e+01, -1.12780483e+01, -8.24639380e+00, 7.90851368e+00, 1.32094938e+01, -8.41980792e+00, -2.41604861e+01, -2.49299721e+01, -4.75880924e+00, 1.34787920e+01, 5.03863481e+00, -8.97469679e+00, -1.22057507e+01, 1.35071090e+01, 1.18731237e+01, -1.01224416e+00, -1.32049942e+00, -6.41030356e+00, -2.79233536e-01, 3.41297173e+00, 1.52804450e+01, 2.24045258e+01, 2.35442009e+00, -7.83038959e+00, 1.38872842e+00, 2.31142280e+01, 5.62018416e+00, 1.32321353e+01, 1.61190312e+01, -7.99750772e+00, 2.47329235e+01, 2.99046960e+00, -7.52167205e+00, 9.31150040e+00, 1.85425255e+01, -5.73424850e+00, -1.35803709e+01, -1.13989175e+01, 9.09572406e+00, 2.58539924e+00, -9.73155133e-01, -8.24384838e+00, 1.34174630e+01, 1.00695986e+01, 7.08746729e+00, -2.55284065e+00, 2.80098530e+00, 4.89980331e-01, -1.98146423e+01, 2.45098617e+00, -9.34430495e+00, 1.39992201e+00, -8.23031878e-01, 7.04671887e+00, -1.20030632e+01, -5.96453807e+00, 3.80323774e+00, -1.08041340e+01, -2.54756327e+01, -9.64829324e+00, 8.56316072e+00, 7.87777571e+00, -3.48107719e+00, 6.00936957e+00, -3.92726875e+00, -2.24542404e+01, -2.50541663e-01, 1.28709204e+01, -1.90364110e-01, 4.38171109e+00, 8.37552522e+00, -1.22853166e+01, 2.45766441e+00, 2.12034850e+01, -7.39163836e+00, 3.87370083e+00, -6.60267251e+00, 9.29828245e-01, 7.43403568e+00, 6.92537467e+00, -6.06421004e+00, -1.83666365e+01, 1.37019947e+01, 2.85835474e+00, -3.23461788e+00, 7.62592572e+00, -1.86596883e+01, 1.21825710e+01, 6.14333088e-01, -2.59078862e+00, 5.94059066e+00, 5.99868541e+00, 1.78076695e+00, -4.72083101e+00, 2.84587542e+00, 1.14185885e+01, 1.75735102e-01, 3.60839658e+00, -2.82040776e+00, 4.11025685e+00, 5.28636236e+00, 1.14406542e+01, -6.43790455e+00, 1.12420444e+00, -1.11169429e+01, 6.59961467e-01, -1.93007169e+00, -1.29782106e+01, 3.36905409e+00, -1.07623785e+01, -1.08202458e+01, 3.57321271e+00, 1.55322239e+01, -8.10596327e+00, 1.75778609e+01, -1.46481689e+00, 2.30515822e+01, -4.47478729e+00, 1.45930648e+01, 7.17670003e+00, -1.22691408e+01, -2.25082987e-01, 9.64364239e+00, -3.95974123e+00, -7.06789811e+00, -5.39158160e+00, -4.99205055e+00, -2.63411159e+01, 2.75322126e+00, -1.16325855e+01, -1.03390586e+01, 9.18850565e-01, 3.99389082e+00, -2.09386018e+00, -2.32049193e+00, 6.51660891e+00, 1.39671594e+01, -2.30382143e+00, -1.71936664e+01, -8.42643183e+00, -1.41545525e+01, 9.03041457e-01, -1.06732036e+01, 1.17100616e+00, 1.97287305e+01, 2.84652167e-01, 4.35293743e+00, 5.73161280e+00, -3.64326596e+00, -2.80847763e+00, -9.65796420e-01, 8.37606536e+00, 5.99515904e+00, 1.16746941e+00, 3.51946228e+00, -1.29349055e+01, 1.90433142e+00, 1.65610885e+00, -2.03067878e+00, -4.17284045e+00, -3.23193712e+00, -4.14257276e+00, 8.50423117e+00, 2.07807961e+00, -2.40150541e+01, -1.39302731e+01, -8.29221147e+00, 1.33580769e+01, -1.23223171e+01, 5.50807979e+00, -1.46699107e+00, 1.43390692e+01, -7.20630611e+00, 6.56103398e+00, -5.63754091e-01, -2.71945076e+00, 3.30539443e+00, 2.71634527e+01, -7.94250109e+00, 2.09921274e+01, -1.85160549e+01, -3.03062431e+00, 3.90359134e+00, 6.05662873e+00, -2.19637965e+00, 4.05427930e+00, 9.37969043e+00, -1.68569834e+00, 1.00312218e+00, -1.18514331e+01, 2.02115155e+00, -1.32824196e+01, -1.47055812e+01, -4.07024782e+00, -5.88531876e+00, -2.46481644e+00, -2.99262980e+00, 7.87940573e-01, -1.32593424e+01, -2.66280410e+00, 1.48102398e+01, -1.09433138e+01, -7.47946544e+00, 3.64373472e+00, -1.43578113e+01, -6.10691266e+00, -7.18434341e+00, 3.74208957e+00, -6.84120764e+00, -7.55761697e-01, -7.64799796e+00, 7.57214073e+00, 6.54684521e+00, -5.01682589e+00, 8.32200965e+00, 1.20624109e+01, 1.03430218e+00, -4.16551123e+00, 1.77712404e+00, 6.80820878e+00, -1.40211504e+00, -1.01230839e+01, -1.23162461e+01, -1.94684781e+01, 9.06430136e+00, 9.93365673e+00, 2.29117461e+00, 9.90249734e+00, -1.54710326e+01, 1.10641311e+01, -2.17011398e+00, -2.61703037e+00, -1.39234548e+01, 3.91608627e+00, -3.49805395e+00, 3.55217620e+00, 5.96252315e+00, 1.74067067e+01, -9.12449519e+00, 1.98676508e+00, -1.18736702e+00, -1.16440099e+01, 4.32403042e+00, -1.68856958e+01, 3.17711410e+00, 1.62947629e+00, 2.15078224e+01, -1.07648182e+01, -2.27177643e+00, -1.04365853e+01, 1.81701634e+01, -1.41910213e+01, 6.11056897e+00, -1.60459519e+01, -8.21006787e+00, -1.31651742e+01, -2.06798527e+01, -1.39646929e+01, 3.33339733e+00, 3.04992530e+01, 1.27437960e+01, 4.24251069e+00, 1.33364158e+01, 1.64329912e+00, -1.40468670e+01, 2.44611729e+00, -3.67863318e+00, 1.48305416e+01, 1.28640715e+00, -1.04333108e+01, -2.79274538e+01, -1.96779053e-01, -6.85236198e+00, 1.74306399e-01, 1.24051441e+01, -1.37465513e+00, -9.64438805e-03, -2.23903737e+01, -1.57692965e+01, -8.60162179e+00, 2.46807544e+00, -1.11535420e+01, 5.87742301e+00, -1.49719200e+01, -2.13898514e+01, -6.21872404e+00, -1.37514531e+01, 7.15248933e-01, -1.16025815e+01, 2.27057665e+00, 3.47117170e+00, -9.27706811e+00, -5.74853847e+00, -6.51985811e+00, 3.19761312e+00, -7.83324024e-01, 2.03362701e+00, -1.06118880e+01, -1.28712820e+00, -1.55885620e+01, -4.32104194e+00, -2.02879849e+01, -1.93060551e+01, 1.81344897e+00, -6.52634723e-01, 1.87566189e+00, 1.08600046e+01, -1.39197941e+01, 1.28905546e+01, 5.32467004e+00, -3.66792469e+00, -1.20869841e+01, 7.24897303e+00, 1.09456764e+01, -2.61881171e+00, -3.59755421e+00, 2.45704500e+01, -1.10078101e+01, 9.26756034e+00, 2.28549370e+00, 2.57059396e+01, -2.54334559e+01, -1.26070749e+01, 1.07489984e+01, -1.37070595e+01, 3.73355649e+00, -2.26291493e+00, 4.29634674e+00, -4.22229930e+00, 6.56404262e+00, 6.74540874e+00, -4.53125923e+00, 1.25735201e+01, -3.40278018e+00, -1.48269505e+01, 1.36954097e+01, -8.06249166e+00, 1.12538049e+01, 1.38402037e+01, -8.69827733e+00, -2.92135312e-01, 1.32332103e+01, 4.73637940e+00, 7.16876626e+00, 2.31614738e+01, 1.59214783e+01, -1.04379614e+01, -1.51202591e+01, -9.63574298e+00, 1.37273252e+01, -5.34972693e+00, -1.32048814e+01, 8.22668353e+00, 2.80237845e+00, -1.66008254e+01, -2.68882792e+00, 7.83770755e+00, -5.44401193e+00, 7.32366021e+00, -3.05391292e+00, 1.14413947e+01, -2.03338875e+01, 3.77707287e+00, -1.17961784e+01, 1.04165304e+00, 4.16397250e+00, -1.20891679e+00, -2.38177765e+00, 2.83194776e+00, -1.52643173e+01, 1.08631558e+01, -4.92605669e+00, -7.65650738e+00, -1.34876499e+01, -2.28649238e+01, -4.57410858e+00, 2.74550251e+00, 2.74218482e-01, -2.03034500e+01, -8.77550551e-01, 5.99755272e+00, 1.68810316e+01, -8.45349292e+00, 1.49763992e+00, -1.07574034e+01, -1.13157707e+01, 6.51892739e+00, -9.43643490e-01, -1.72846216e+00, -1.29509968e+00, -1.29275828e+01, 3.64495982e-01, 3.75560629e+00, 1.19191180e+00, 7.88728649e+00, 9.74326778e+00, -1.23545925e+01, 1.59734609e+00, -1.02358251e+01, 7.72428850e+00, -2.30601465e+00, 4.30860432e+00, -1.77988532e+00, -2.48779398e+01, -9.28989011e+00, -1.10724861e+00, 7.18861678e+00, 6.75932601e+00, 2.00187273e+00, -1.63466964e+00, 1.92160285e+01, -6.99944131e+00, 1.35355307e+01, -7.37858204e+00, 3.26666225e+00, -7.96014336e+00, 7.59861069e+00, -5.56364763e+00, 1.34899848e+01, -1.86809862e+00, -1.25095765e+01, 7.88334934e+00, -5.30206725e-01, -1.15407160e+01, 1.38225864e+01, -2.87625457e+00, 1.14184786e+01, -8.04665555e+00, -9.36164550e-01, 2.07395899e+01, -6.57120687e+00, 4.20917477e+00, 2.09294082e+01, -2.90660984e+00, -6.05612656e+00, 5.44477766e+00, 3.48610502e+00, -6.60534235e+00, -7.11304095e+00, 5.37281935e+00, -8.99827164e+00, -1.06498210e+01, 4.19712932e+00, 1.17412320e+00, 1.80637524e+01, 7.05075623e+00, 1.04485882e+01, 3.54234429e+00, 5.79906673e+00, -1.33351290e+01, 1.41423160e+01, -8.95182192e+00, -4.31568263e+00, 1.04177861e+01, 1.26616312e+01, 7.67378649e+00, -2.16982781e+01, -1.00816551e+01, 2.35518737e+01, 6.86530772e+00, 6.40697601e+00, -1.99322748e+00, 3.98246319e+00, 2.05288233e+01, 1.39854696e+01, -1.17054961e+01, 6.04978021e+00, 4.40863513e+00, 9.24759948e+00, 4.49112700e-01, -3.73650634e+00, -1.29550453e+01, 1.66737979e+01, -1.29217769e+01, 4.37659712e+00, -7.17431367e+00, 1.34761041e+01, 1.28740431e+01, 3.65019299e+00, -4.51121741e+00, 1.09273790e+00, -2.37040194e+00, -1.94368144e+00, -1.53354225e+00, 1.61068054e+01, 3.09931942e+00, -1.80617615e+01, 2.15583174e+00, -7.61688834e+00, -3.91699995e+00, -4.20206013e+00, 3.44904518e+00, 7.70283094e+00, -1.12672523e+01, 3.85767141e+00, 1.25304623e+00, -9.85203195e+00, -1.08827795e-01, -4.73437011e+00, 4.81716938e+00, 6.30232158e+00, -1.21726982e+01, 8.93600976e+00, -1.37533124e+01, -6.18407517e+00, -7.26695102e+00, -1.06277951e+01, 2.66285025e+00, -7.54966489e+00, 1.51955994e+01, 1.78114541e-01, -3.36344782e+00, -4.88392448e+00, 1.75039304e+01, 6.78119997e+00, -8.82927438e+00, -8.47610904e+00, -1.21456751e+01, 9.63552622e+00, -1.98378110e+01, -9.66024838e+00, 1.38543827e+00, -1.30898394e+01, 7.68208435e+00, -8.02173899e+00, 6.86812488e+00, -1.03544567e+01, 2.17095337e+00, 1.21861988e+01, -1.87335019e+00, 9.94388673e+00, 2.24795481e+00, -1.11808209e+01, -1.02444828e+01, 2.83471831e+00, 4.82592021e-01, -7.06726391e+00, 1.08942544e+01, 1.24659810e+00, -1.40673697e+01, 1.32327038e+01, -1.57108329e+01, -1.75334861e+01, 7.08369754e+00, 1.40864235e+01, -2.65188311e+01, 6.32079714e+00, 3.59577557e+00, -1.29752581e+01, 1.77481215e+01, -3.07296471e+00, 1.08750989e+01, 8.52826097e-01, 1.40369651e+01, -6.48204908e+00, 4.53679735e+00, 1.62885653e+00, 2.16431798e+00, -1.50502935e+00, 1.41826273e+01, 8.39262368e+00, 2.79261297e+00, -1.65216944e+01, 1.19925631e+01, -2.00547315e+00, -2.27278973e+00, -7.33755176e+00, -1.29671900e+01, 1.17336047e+01, -6.62628390e-03, 3.27095879e+00, 2.01840757e+00, 2.87668878e+01, 6.09763446e+00, 6.86624271e+00, -3.53508260e+00, -5.70502920e+00, -8.12630657e+00, -7.76413167e+00, -2.36285877e+01, 4.51707643e+00, 2.24780474e+00, 9.73766016e+00, 7.56308848e+00, 7.92312868e+00, 2.62134143e+01, 1.48966272e+01, -2.51127913e-01, -1.08603351e+01, 1.38586257e+01, 3.93853893e+00, -5.13204830e+00, -8.47547768e+00, -5.28988332e-01, 7.88877668e+00, -6.34469600e+00, -1.11059594e+01, -9.19961077e+00, 2.50666560e+01, 1.63126039e+01, -2.76726120e+00, -9.66728412e+00, 3.03591034e+00, -1.08992066e+01, -1.30086717e+01, 1.69221866e+01, 2.20256417e+01, -1.10545329e+01, 3.50376330e-01, -6.90355188e+00, 9.44602995e+00, -2.91875937e+00, 1.66011194e+01, 2.63996799e+01, -1.27576930e+01, -1.02189666e+01, -9.07092475e-01, 1.99954293e+00, -2.30263785e+01, -9.95310420e+00, 1.74388607e+01, -7.43659080e+00, -1.86104996e+00, -6.26792853e+00, -1.77563285e+00, -1.10975111e+01, -8.36704413e+00, 5.55223950e+00, 1.01638548e+00, -3.41420376e+00, 2.89375811e+01, -8.30050294e+00, 6.80054156e-01, -1.12570467e+01, -2.14337479e+00, -1.15525381e+01, 9.40782045e+00, 4.70689725e+00, 6.98255237e+00, -1.80286860e+01, 9.50598407e+00, 1.14907317e+01, -7.21410598e+00, 5.95980631e+00, 1.47695538e+01, 2.18677747e+01, -3.56527529e+00, -2.88338579e+00, -2.88919962e+01, -1.50428808e+01, 5.75214104e+00, -7.09680112e+00, 7.91132905e+00, 5.62263395e+00, 9.59879753e+00, -2.35489899e+01, -1.32419311e+01, -5.56360877e+00, -7.88479117e+00, -1.94414526e+00, 1.14007530e+01, -3.20676751e+00, 6.76766226e+00, 7.37185904e-02, -2.45481520e+00, 8.95722055e+00, -5.53491385e+00, -5.28741174e-01, 4.87937147e+00, 6.81174810e+00, 3.26613095e+00, -1.23257812e+00, -1.52468640e+01, 2.67082125e+00, 1.22283176e+01, 2.39229490e+00, -1.07033504e+01, -4.02452952e+00, 2.07756248e+01, 1.99521183e+01, 1.98993155e+01, -4.56218215e+00, 1.02347059e+01, -3.10551088e-01, 1.21285553e+01, 7.46433607e+00, 3.58982732e-01, 3.70573663e+00, 2.62001495e+00, 6.71785709e+00, -1.13768630e+01, 1.69302649e+01, 5.68573876e+00, 2.59553612e+00, 2.02231808e+01, -5.01073391e+00, -4.16930718e+00, 4.48818482e+00, 2.69112041e+01, 7.77842750e+00, 3.99998908e+00, 8.71986337e+00, 1.76378193e+00, 6.05153549e+00, 1.07711797e+00, -3.83897837e+00, 5.50431240e+00, 5.29869607e+00, 6.55872356e+00, -7.13667306e+00, 5.30684088e+00, -7.45847529e+00, -3.26229397e+00, -3.01913257e+00, -2.16712947e+01, -3.77654393e+00, 2.37469409e+01, 4.95151838e+00, -1.42615733e+00, -5.61376712e-01, -8.61662136e+00, -8.73175364e+00, -1.08036977e+01, 1.51908379e-01, -5.90683887e-01, 5.93659778e+00, 1.11423454e+00, -6.79702447e-01, -5.76789047e+00, -1.32984564e+01, -1.68769955e+00, 6.04105538e-02, -6.66837404e+00, -9.16388676e+00, -1.54550932e+01]), 'y': array([1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0]), 'pred': array([1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0])}
1 2 | r = generate_classification_results(p_pos=0.9, pred=1) r |
{'acc': 0.896, 'bacc': np.float64(0.5), 'confusion': array([[ 0, 104], [ 0, 896]]), 'n': 1000, 'seed': None, 'noise_sd': 1, 'signal_sd': 10, 'p_pos': 0.9, 'shuffle': False, 'thresh': np.float64(-11.641292591387709), 'clf': None, 'X': array([-9.91180680e+00, -4.86661300e+00, 2.52586964e+01, 7.45769636e+00, 5.23023246e+00, -1.69371714e+01, 1.73295178e+01, 1.46053040e+00, 1.15121414e+01, -1.78255088e+01, 8.66810002e+00, -3.51403145e+00, 6.67156100e-01, -9.43157850e-01, 8.84251399e+00, 8.55207151e-01, -2.86461710e-01, 8.68657329e+00, 8.06014793e+00, 1.24574842e+01, -3.33246436e-01, -2.88697965e+00, -2.71651022e+00, 1.48399972e+00, -3.38700076e+00, 7.14679145e+00, 8.23363045e+00, -5.71729798e+00, -2.20369064e+00, 1.59119742e+01, -8.46544617e+00, -4.55688333e-01, -2.28000652e-01, -3.28379111e+00, 4.01081987e+00, 2.11624336e+01, 6.85509782e+00, -3.90060595e+00, 1.24550035e+01, 2.41014519e+00, -7.22182138e-01, -1.15210454e+01, 5.55135970e+00, 3.67328173e+00, -1.96599820e+00, -5.17721380e+00, -3.17728419e+00, -2.94654153e+00, 6.54523527e+00, -1.82276131e+01, 2.29074223e+01, 5.26598532e+00, -1.42734627e+01, 2.72174110e+00, 5.58615374e-01, 3.30534508e+00, -8.86569168e-01, 1.11110135e+01, -6.37406775e+00, 5.82980785e+00, -1.22654481e+01, 7.56696518e+00, -1.30876931e-01, 4.28379842e+00, 4.91736534e+00, 5.78474395e+00, 7.92467921e+00, 2.90358502e+00, 1.56717582e+01, 2.25884049e+01, -1.12959415e+01, 8.41889027e+00, 7.38044387e+00, -2.49120424e+00, -3.86038027e+00, -1.17156215e+00, 3.27298481e+00, 5.85387231e+00, 1.11233029e+01, -1.84296808e+01, -5.93254059e-01, 9.31427968e+00, 1.19435802e+01, 9.43825542e+00, -7.76419387e+00, -3.81288547e+00, 5.68912312e+00, 4.95686073e+00, 1.36127212e+01, 1.46329288e+01, 2.06020208e+00, -1.04183740e+01, -2.13631497e+00, 1.41229355e+01, 1.22392523e+00, -4.91386813e+00, 9.17586615e+00, -5.82540646e+00, -1.93462934e+01, 8.78988112e+00, -1.24215210e-01, 1.43787112e+01, -5.41204705e+00, -4.86424359e+00, 2.19471518e+00, -5.21884666e+00, -1.11413549e-01, -9.19301973e-01, -5.58021709e+00, -2.01970104e+00, -8.89530099e-01, 4.35487119e-01, -4.98928801e+00, 1.19344320e+01, -2.80512512e+01, -2.88261648e+00, 4.70336952e+00, -2.42514134e+01, 4.44837477e+00, 2.21434892e+00, -5.14862162e+00, 8.06970979e+00, 1.94499385e+01, 1.65072339e+00, -4.27204436e+00, -6.23414364e+00, -1.05793350e+01, 6.48740721e+00, 1.09926832e+00, 2.05445558e+00, -7.34208245e+00, -2.35638234e+00, 2.15682841e+00, 4.85402186e+00, -1.10581368e+00, -2.70308621e+01, -9.87787074e+00, -7.48679124e+00, 1.96469175e+00, -8.00935937e+00, -1.40888285e+01, -1.75942654e+00, -1.03266156e+01, -3.69709850e+00, 1.39805045e+01, -2.11265408e-01, 2.87503489e+00, 9.30083742e-01, 7.06607793e+00, -6.67672864e+00, -4.18043797e+00, 1.01051761e+01, 5.35357718e+00, 3.82854987e-01, -1.84027632e+01, -3.57783251e+00, 1.10677411e+01, 1.03533748e+01, -3.46879420e+00, 1.20178796e+01, -1.00082369e+01, 1.73408834e+01, -8.71225360e+00, 9.68233750e+00, -3.62272614e+00, -4.46621285e+00, -7.64881704e-01, -6.70194355e+00, -5.12577323e+00, 3.01487701e+00, 6.09225832e+00, -1.55359937e+00, 7.25879521e+00, -3.01002519e+00, 8.89505734e+00, -2.25597062e+01, -2.30285574e+01, 4.32835468e+00, 5.54603437e+00, -7.70862407e+00, 3.55306621e+00, 3.00082537e+00, 3.23955816e+00, -4.10439612e+00, 1.27487897e+01, -1.53446690e+01, 4.38722643e+00, 4.21387809e+00, 8.09681165e+00, -7.29944414e+00, 3.36541538e+00, -3.34234446e+00, -6.53857364e+00, 3.66050419e+00, -5.55734939e+00, -3.31022599e+01, 9.72045037e+00, 1.48362556e+00, 1.31931641e+01, 4.56155231e+00, -4.86751078e+00, 1.05394225e+01, 2.29741408e+01, -5.64307522e+00, 1.50141554e+01, 2.59064985e+00, 4.37400611e+00, -1.43154648e+01, 7.80794045e+00, 1.76933178e+00, -4.45536142e+00, 3.65647551e+00, 7.71009509e+00, 9.27591988e+00, 1.17637022e+01, -7.00071620e+00, 3.69522935e+00, 2.06388963e+00, 4.04335177e+00, 9.92285690e+00, -5.17655985e+00, 8.74845295e+00, 1.48701826e+01, 1.26682575e+01, -4.92985132e+00, 1.28246294e+00, 2.24060752e+01, 3.05344352e-01, -6.37207354e+00, 1.10284916e+01, 1.23894916e+01, 9.86783303e+00, 1.31123275e+01, 2.88842807e+00, -8.04760898e+00, -1.62027798e+01, 4.42613906e+00, 8.02740326e+00, -8.65266050e+00, 5.04490351e+00, -3.34507801e+00, 1.24557151e+01, -1.43282637e+01, 1.85070620e+01, 8.91097298e+00, -2.89467126e+00, 2.16290107e+00, 4.25458300e+00, 1.70552429e-01, 2.25748041e+00, -1.25373263e+01, 6.96041013e+00, -1.41049201e+01, -7.75673079e+00, 6.23270202e+00, -8.73193263e+00, -4.16137311e+00, 3.71916387e+00, -1.25194827e+00, -2.14360402e+01, 9.89962825e+00, 2.00545392e+01, -2.41743374e+00, 2.12234837e+00, 1.11910700e-01, -9.25672195e+00, -1.48106938e+01, 1.00460674e+01, 9.08414143e-01, -5.34883742e+00, -5.16932677e+00, 6.61260951e+00, -1.41955834e+01, 2.03085672e+01, -5.20365881e-01, 8.70282133e+00, -1.48762048e+01, 2.44781233e+01, 1.24770321e+01, 1.74872642e+00, -1.40395255e+01, 1.74460181e+01, -9.21677768e-01, 5.16682058e+00, -1.11721767e+01, 1.16419058e+01, -1.18150210e+00, -1.17560743e+01, -2.40494312e+00, -7.75601627e+00, -1.12863377e+00, -3.80187376e+00, -2.27976529e+01, -1.61695380e+01, 3.70685487e+00, 2.73896902e-01, 2.41009659e+00, -1.58964775e+01, -1.27928583e-01, -2.30469046e+01, 6.29201705e+00, -8.09640463e+00, 6.57136825e+00, 3.55893365e+00, -4.85124741e-01, 1.17960413e+01, -3.40937781e+00, 1.05078641e+01, -4.74522586e-01, 6.38710886e+00, 5.50853007e+00, -5.86931901e+00, -3.37626473e+00, -1.94051473e+00, -2.07906749e+00, 2.78042657e+01, -4.82736556e+00, -5.07971396e+00, 1.67918569e+01, -1.30801060e+01, 1.71635962e+00, 3.15863363e+00, 6.36626248e+00, -2.34826105e+01, -4.42968692e+00, 4.21381851e+00, 8.15408926e+00, -1.51068395e+01, -1.91463829e+01, -8.12862055e+00, 9.91420000e-01, 2.36309432e+01, -6.60948903e+00, -1.02137350e+01, 1.08417215e+00, 3.44765990e+00, 7.08021689e+00, 4.13542052e-01, 9.30140356e+00, 2.72887849e+00, 1.39669796e+00, -4.02239250e+00, -2.61453120e+00, -1.98450718e+00, -4.91557547e+00, -1.63154796e+01, 1.28907999e+00, 1.36527850e+01, 1.74654563e+01, -4.81511515e+00, 9.16873495e-01, 5.62270131e+00, -2.01565696e+00, 5.13313204e+00, -9.46092352e+00, -3.84507949e+00, 7.56377956e+00, -6.95166816e+00, 7.70350272e+00, -2.73587365e+00, 5.34779749e+00, 5.62840207e+00, -1.74572194e+01, -7.36286229e-01, -1.28320159e+01, -2.01086241e+00, -1.08199936e+01, 1.00336394e+00, 2.38925540e+01, -8.41998164e+00, 1.27438959e+01, 6.68311240e+00, 3.16079574e+00, 1.05097423e+01, 4.29032242e+00, 5.66301257e+00, 1.95087728e+01, 8.00766404e+00, 3.32908768e+00, 3.48935395e+00, -5.61184665e+00, 1.75095553e+01, 5.45061240e-01, -8.19618886e+00, 6.84935787e+00, 8.23603320e+00, 4.18909329e+00, -1.68602825e+00, 1.67698311e+00, -9.13482813e+00, 7.16502157e+00, 7.13872414e-01, 1.07766632e+00, 7.01939171e+00, -8.32432555e+00, 3.25487170e+00, 2.18323636e+00, 1.28823252e+01, 1.42886797e+00, 1.21169042e+01, 1.75203153e+00, 1.30295238e+01, 2.72709788e+00, 5.26758055e+00, 1.33539720e-01, 4.03232694e+00, -5.79299608e+00, 3.07278740e+00, 9.33147564e+00, 1.45350883e+00, 6.40614248e+00, 5.97359098e+00, -3.11095390e+00, -7.57159779e+00, 3.08248924e-01, 3.88896510e+00, -3.86958199e-01, -1.85301517e+00, -5.13593746e+00, -2.47824658e-01, 8.60878043e+00, 2.05720076e+01, -6.35454509e+00, 6.54279020e+00, 1.61088813e+01, -1.32364940e+01, 3.60799929e+00, -6.99182018e+00, 7.16647904e+00, -4.68386713e+00, -2.17730566e+00, -8.25007964e+00, -1.10796498e+01, -1.67684188e+01, -1.60208755e+00, -1.24928106e+01, 3.17437030e+00, 1.35936682e+01, 9.14712802e-03, 3.90691595e-01, 4.45602039e+00, -5.79148920e+00, 1.58916047e+00, 2.81635916e+00, 6.39977807e+00, 1.09674116e+01, -1.32652909e+01, -8.71056479e+00, -1.38769388e+01, 1.92929333e+01, -6.56537208e+00, -5.44063931e+00, -4.69053012e+00, -1.64421178e+00, 4.59719680e+00, 6.21664590e+00, 1.48109575e+00, 1.47332263e+01, -5.64021991e+00, -1.98505935e+01, 1.76049550e+01, 8.51488070e+00, 3.45750779e+00, 1.15298814e+01, 1.39059284e+01, -3.05307172e+00, 2.02363319e+01, -2.11531644e+01, 2.25287199e+00, -7.19332615e+00, 6.16580219e+00, 6.13582959e+00, 8.39761207e+00, -1.16256431e+01, -3.46407627e+00, 1.07680568e+01, 2.04552990e+01, -2.30058315e+01, 9.03246408e+00, -1.85723024e+00, -4.18255421e+00, -7.29036178e+00, -8.52379918e+00, -5.57708720e+00, 2.89223762e+00, 4.22961396e+00, 7.75878977e+00, -3.78646632e+00, -6.19792426e+00, 9.02107305e+00, -1.77665906e+00, 2.74288139e+01, -2.25435228e+01, -2.18150312e+01, 1.32576859e+01, -2.15297777e+01, -1.21914539e+00, 1.34875411e+01, 2.07401424e+00, -2.42637492e+01, -1.57735341e+01, 5.27465476e+00, -1.56620790e+01, 6.09752072e-01, -1.00648080e+00, -2.82073886e+00, 5.93344711e+00, -9.59454192e+00, 1.26811891e+01, 1.50766439e+01, -1.69141477e+01, -1.52962411e+01, 1.08333871e+00, -1.09283261e+01, -2.22111675e+00, -7.04697309e+00, 8.78064405e+00, 7.16008574e+00, 5.09337135e+00, 8.51148671e+00, 1.29565028e+00, 6.22723954e+00, 3.92064045e+00, 2.15954592e-01, 5.49772710e-01, -6.13604370e+00, -1.72053084e+01, 5.52895315e+00, -9.96397237e+00, -1.52706766e+01, -3.22156199e+00, -1.24292221e+01, -2.61893242e+00, 7.07952816e+00, -3.56443073e+00, -8.35610200e+00, 4.77889017e+00, -2.15468019e-02, 1.82490580e+00, -5.72441857e+00, 6.51181870e+00, -9.35190564e-01, -1.13217365e+01, 4.17308610e+00, -3.29011890e+00, 3.99276869e+00, -6.52471911e-03, 9.79124937e+00, 1.37738530e+01, -2.44826427e+00, 3.81439700e+00, -1.20602806e+01, 1.31401229e+01, 7.88666227e+00, -1.92923014e+00, -7.68441593e+00, -6.84965112e+00, 1.29487053e+01, 4.08272596e+00, 1.24392074e+00, 5.76753063e+00, -3.23906426e-01, -1.40721850e+00, -3.24465475e+00, 6.79936592e+00, 3.21988844e+00, 1.03815054e+01, 2.16385540e+01, -6.86259580e+00, 7.27857970e+00, -9.87904937e+00, -4.50481005e+00, -3.08785679e+00, -4.33717486e+00, 1.37454860e+01, 3.26786195e-01, -4.91855515e+00, -1.54817627e+01, 3.94417559e+00, 6.66606725e+00, 2.82953575e+00, -8.72672257e+00, -2.19497435e+01, -1.69825828e+01, -4.19145681e+00, -1.08049533e+01, 9.37546764e-01, -9.68913673e+00, 7.92387525e+00, 2.29537849e+00, 4.91872641e-02, 1.45228646e+01, -4.99002385e+00, 1.08658332e+01, 1.24821627e+01, -1.12732778e+01, 7.01983955e-02, 2.53325926e+00, 1.44663336e+01, -9.68858753e+00, -2.12105930e+01, -1.42484085e+01, -1.84658002e+01, 5.14551261e+00, 8.11512674e+00, -2.20186872e-01, -2.20822397e+01, -3.27057440e+00, 1.62769729e+01, -1.47243637e+01, 4.75137312e+00, 4.95447514e+00, 8.28740562e+00, 3.47016933e+00, 1.94746052e+01, 5.13117757e+00, 1.11021307e+00, 2.47173114e+01, 2.21700994e+00, 1.17952899e+01, 1.49158754e+01, 3.59921628e+00, 7.03855276e-01, 1.72007887e+01, 1.25409479e+01, 1.37773789e+01, 4.46877865e+00, 1.45447160e+01, -2.70994975e+00, -9.32117382e+00, 2.81528103e+00, -4.32569807e+00, -1.58767439e-01, -8.84355481e+00, 1.46937369e+01, -2.01803672e-01, 1.56663355e-01, -4.73606316e-01, -1.70013942e+01, 4.21381367e+00, 2.66383348e+01, 8.51742240e+00, 1.01171015e+01, -4.99702708e+00, 3.06983597e+01, 6.03391625e+00, -1.44946618e+01, -6.20634987e+00, 3.29652708e+00, 3.51535855e+00, 1.84874817e+01, -4.68993415e+00, 1.33883894e+01, 8.55652028e+00, 2.11976926e-01, 2.94505927e+00, -4.07870849e+00, -9.89327473e+00, 9.96000126e-02, -9.31981465e+00, 5.56067762e+00, 6.42222343e+00, -4.42952618e+00, 9.65446889e+00, 1.82412033e+01, 5.42004283e+00, 2.48697266e+00, -6.54049862e+00, -1.97117327e+01, 9.87703882e+00, 3.46106342e+00, -4.58538069e+00, -6.36446420e+00, 2.38480015e+00, -1.18239609e+01, 1.67953177e+01, 6.06416240e+00, -1.56458095e+00, -2.08894278e+00, -6.15578762e+00, -4.27228379e+00, -8.48891556e-01, 8.87360844e-01, -2.58201246e+00, -1.59779673e+00, -3.43733204e+00, 8.45596815e+00, -1.16343377e+01, 1.80050880e+01, 4.96666363e+00, -8.17096949e+00, -1.60775258e+01, 6.53873748e+00, -6.98526446e+00, 5.63331592e+00, -2.36510532e+00, -4.74076529e+00, -3.38770810e+00, -1.37881055e+01, -2.34776066e-02, -1.40708278e+01, 2.94885748e+00, 8.19868319e+00, 1.10550703e+01, -7.95092138e+00, 6.07413759e+00, 8.16634068e-02, 1.86185721e+00, -2.56200610e-01, -2.40913567e+01, -6.67573814e+00, 3.62778319e+00, -1.10924785e+01, -3.92617103e-01, -7.47983904e+00, -5.69203842e+00, 1.02872020e+01, 1.20128600e+01, -8.12303477e-02, -5.17991458e+00, 4.20831460e+00, -1.75169837e+01, 6.12172167e+00, 6.68738505e+00, 1.86485997e+01, -3.37937541e+00, -2.25823410e+01, -2.04410565e+01, 7.93041754e+00, -3.34944315e+00, 5.12674541e+00, 6.56391741e-01, -3.12270631e+00, 4.84352521e+00, -1.16380673e+01, 2.20741620e+00, -6.07174208e+00, 1.07876670e+01, -1.03748473e+00, -1.22598539e+01, -1.01858807e+01, -2.42354404e+01, 2.56099232e+00, 9.28923218e+00, 1.12261519e+01, 1.23007999e+01, -5.36857312e+00, -1.84802201e+01, -6.28294990e+00, 6.91553154e+00, 1.53849043e+01, -4.02340645e+00, -9.36332064e-02, -7.97568635e+00, 3.91799459e+00, 1.06790435e+01, 7.81212551e+00, 2.66456221e+01, 1.86680880e+01, 1.59503250e+01, 1.31513627e+01, 8.02556772e+00, -1.97766307e+00, -4.39761086e+00, -8.80040429e+00, -1.62671376e+01, -1.64291677e+00, -7.31106018e+00, 5.05830577e+00, -3.15434153e+00, -4.74587203e+00, -5.84442130e+00, 9.73559578e+00, 2.44728687e+00, 2.28970828e+01, 2.96655874e+00, 3.41481502e+00, -1.81393133e+01, 8.62777146e+00, 3.92201500e+00, -7.18128913e+00, -1.19889940e+01, 4.58878166e+00, -6.77786315e+00, 5.40571260e+00, -4.31408847e+00, -3.05737656e+01, -1.04245315e+01, 6.27153926e+00, -1.74392922e+01, -3.53145842e+00, 5.30677353e+00, -4.22332679e+00, 7.16065419e+00, -1.82680937e+00, 1.60696342e+00, -1.21822062e+01, -6.19497685e+00, 1.56201231e+01, -1.02583975e+00, 8.21362813e-01, 1.07713663e+01, 1.44492694e+00, 5.62973858e+00, -1.71298859e+01, -1.84615476e+00, 5.58574174e-01, 2.36364697e+00, -1.15355952e+00, -5.43402120e+00, -1.14450996e+00, 4.49380065e+00, 3.88893403e+00, -6.00654308e+00, 1.24981860e+01, 6.73158005e+00, -1.15798098e+01, 4.32672328e+00, 5.63314248e+00, -5.47133171e+00, 5.16199473e+00, -6.67001417e+00, 1.71668304e+01, 6.91731474e+00, 1.15408327e+01, -8.44311256e+00, -2.60350879e+00, -4.11514323e+00, 6.98050907e+00, 6.11078361e+00, -8.82317445e+00, -1.93966124e+01, 5.36619531e+00, 9.91694675e+00, 3.84727538e+00, -1.38639091e+00, -6.81346070e+00, 1.26949406e+01, 7.23263746e-02, 8.26664180e+00, 6.87724026e+00, 1.42407492e+00, 1.55379430e+01, -1.16703204e+01, -1.10450928e+01, -5.16273394e+00, -1.06786805e+01, -6.20892459e+00, -7.38547178e+00, 1.39286897e+01, -6.39885018e+00, 1.91946717e+01, 6.92956594e+00, -9.44791736e+00, 1.62197076e+01, 2.30438501e+01, -5.41191945e+00, 6.81033625e+00, -3.59237608e+00, 2.16860178e+00, 4.71395960e+00, -5.31768497e-01, 6.32377140e+00, -6.17647846e+00, -1.05399701e+01, -2.06274669e-01, -8.01288956e+00, 1.51291626e+01, 8.11019983e-01, -1.43854539e+01, 7.02993304e+00, -1.60759262e+01, -2.03590847e+00, -1.71397840e-01, 6.35112731e+00, 7.20546008e+00, -1.61494574e+01, 6.50503032e-01, -9.09934475e+00, 1.89151823e+01, -2.07391194e+00, 2.52325008e+00, -1.08323444e+00, 7.01459080e+00, -3.76939363e+00, 1.34616032e+01, -3.59158462e+00, -4.42582871e+00, -3.95595016e+00, 5.61608393e+00, 2.99850518e+00, 7.51241602e+00, -8.56639406e+00, -1.96832561e+01, -3.84894035e+00, 1.31073230e+01, -3.81344034e+00, 5.73331273e+00, 4.30442029e+00, 2.94917879e+00, -3.09838144e+00, 1.22658865e+01, 2.11192934e+00, 2.99701117e+00, -7.54493461e-01, 1.34787576e+00, 2.85038952e+00, 1.36777684e+01, -6.44674900e+00, 2.04105982e+01, 8.26290922e+00, 5.24888403e-01, 4.51983714e+00, 5.00517294e+00, -8.67236330e-01, -1.06941871e+01, -1.01826351e+01, 2.95966706e+00, -1.28019433e+01, 1.75328227e+01, 5.41549722e+00, 1.74335504e+00, 7.73217728e+00, -2.86254151e+00, 1.96492419e+01, -1.48407353e+01, 6.01427032e+00, -1.56928312e-01, 8.76225954e+00, 9.48267157e+00, -1.55361658e+01, 9.28554959e+00, 9.90209271e+00, 6.84101959e+00, -7.96520086e+00, -2.48782438e+01, -2.78647033e+00, 5.59333027e+00, 1.95920934e+01, -8.99523670e+00, 1.27493917e+01, -9.69266765e+00, 1.28030936e+01, 5.98798068e-01, 9.73688393e+00, -6.34213091e+00, -3.43034305e+00, 1.30776986e+00, -1.14713858e+01, -1.18520855e+01, -6.59248970e+00, -9.21753526e-01, 8.46198232e+00, -1.60882144e+00, 4.37895727e+00, 1.44248737e+01, 1.31845529e+01, 1.13425870e+01, -1.44589665e+01, -4.34642727e+00, 4.39320657e+00, -5.39487642e+00, 7.31668218e-01, 1.17925364e+01, 1.90199742e+00, 4.70926352e+00, 8.62669009e+00, 1.31506538e+01, -1.88034072e+00, 6.16581127e-01, -5.71332954e+00, -1.07962202e+01, 2.16952395e+00, 5.71740769e+00, -6.00321159e+00, -1.12031067e+01, 6.36859702e-01, 3.41520194e+00, -1.15498976e+01, -2.69705971e+00, -6.43517720e+00, 5.28717490e+00, 9.91024140e+00, 6.90279952e+00, -1.19823781e+01, 1.51142056e+01, 5.02094503e+00, -1.05162980e+01, 1.91051532e+01, -8.62682238e+00, -7.14013286e-01, 9.58194927e+00, -5.42413060e+00, 1.90104232e+01, -7.06842859e+00, 1.51504230e+01, 5.44942193e+00, 1.52504760e-01, 2.30548307e+01, -3.91445792e+00]), 'y': array([1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]), 'pred': array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.])}
1 2 3 4 5 | from sklearn.naive_bayes import GaussianNB clf = GaussianNB() r = generate_classification_results(p_pos=0.8, clf=clf, noise_sd=10) r |
{'acc': 0.805, 'bacc': np.float64(0.7218452237969413), 'confusion': array([[148, 128], [ 67, 657]]), 'n': 1000, 'seed': None, 'noise_sd': 10, 'signal_sd': 10, 'p_pos': 0.8, 'shuffle': False, 'thresh': np.float64(-9.033025990136332), 'clf': LogisticRegression(), 'X': array([ 6.92884901e+00, -8.54685728e+00, 2.73355295e+01, -3.30980315e+00, -1.09530402e+01, 1.87713081e+01, -4.27771441e+00, -4.24332374e+00, 2.43273755e+01, 1.12690163e+01, 3.55285060e+00, -2.46919663e+00, -1.46259624e+01, 1.21136474e+01, 3.02175217e+00, -1.93156454e+00, 1.83551239e+01, 4.77806478e+00, 1.13148112e+01, -1.06678999e+01, -2.15906341e+00, 8.09996144e+00, 5.24032510e+00, 6.52227010e+00, 2.78628726e+00, -1.44382740e+00, 4.60704077e+00, -7.09372421e+00, -4.83274788e-01, -2.66784320e+01, -1.23718454e+01, -5.96679885e+00, 3.30606137e+00, -5.74056474e+00, 1.04736830e+01, 8.79095203e-01, 6.34975221e+00, -7.74186361e+00, -9.12932014e+00, 1.01838899e+01, 1.59046608e+01, 1.88148966e+01, 8.64023785e-01, 1.05010045e+01, 1.39500851e+01, 9.45406220e+00, -7.32379797e+00, 9.20281511e+00, -4.01782621e+00, -1.31565909e+01, 1.19270266e+00, -7.91349328e+00, 1.29437581e+00, -2.75527270e+01, 1.81109134e+01, -1.29082728e+01, 4.06230860e+00, -1.98180693e+00, 1.25082005e+00, -7.99859326e+00, 1.78360642e+00, 2.12872915e+01, 5.77571092e+00, -1.28837392e+00, -1.34385711e+00, -1.11612481e+01, 4.02125831e-01, 7.26792987e-01, -2.03922008e+00, -2.51239360e-01, 2.87315090e+00, -1.35869033e+01, 1.23833623e+01, 1.22593693e+01, 7.96471957e-01, -1.63023442e+01, 4.40900850e+00, 2.90238030e+00, -3.43618448e+00, 1.14128587e+01, 3.04403563e+00, 1.30805478e+01, -6.63821456e+00, -2.40482695e+00, 3.85932561e-01, -1.04280174e+01, 1.09101705e+01, -1.05075639e+01, -1.89366598e+01, -7.67373384e+00, 2.01269784e-01, -2.36336737e+01, 1.88016425e+01, 8.99501188e+00, -4.77686049e+00, -1.92311148e+01, -3.55816266e+00, -8.05193480e+00, -3.36469530e+00, 1.58128385e+01, -3.00058996e+00, 4.57129624e+00, -3.67623745e+00, -1.22725654e+01, 1.52409014e+00, 1.14261329e+00, 1.49429035e+00, 1.79421530e+01, -1.25926457e+01, -2.37644374e+00, -7.73492294e-01, 9.27396835e-01, -8.39161421e+00, -1.16122468e+01, -1.59744164e+00, 2.78714438e+00, 1.58247215e+01, 4.44014649e+00, -1.03597065e+01, -1.39167146e+01, 8.08352234e-01, -1.91266168e+01, -2.77544871e+00, -2.88565140e+00, -3.95311471e+00, -2.16755577e+00, -1.98085012e+00, 2.21247765e-01, -4.91841496e+00, -2.06148283e+01, 3.61044136e+00, -1.08050761e+01, -9.70515848e-01, -3.88794496e+00, 1.89058690e+00, -2.59498084e+00, -4.00183513e+00, -2.84927582e+00, 1.91276571e+00, 2.20464911e+01, 5.54606289e+00, -3.14882415e-01, -5.18076311e+00, -1.52475275e+00, 9.12954190e-01, -1.72719759e+00, -1.13469100e+01, -1.68411972e+01, 4.80064715e+00, -7.76769391e+00, -5.52740863e+00, 1.32004319e+01, 6.05684259e+00, 9.91459639e+00, -6.99611542e+00, 1.34477365e+01, -4.05523081e+00, -8.66530037e+00, -2.41221283e+01, -1.90830611e-01, 6.21694428e+00, -5.63457512e+00, 8.50713231e+00, -1.31417423e+01, -6.70238915e+00, 4.25800810e+00, -1.85729816e+00, -1.56474290e+01, 5.47517587e+00, 5.06836803e+00, 4.09511910e+00, 4.51206877e+00, -5.99236717e+00, -6.59993496e+00, 1.65371697e+00, -3.71160386e+00, -6.47169891e+00, 5.87290114e+00, -2.01316747e+01, -9.31527534e+00, -4.50983933e+00, -1.92370090e+00, 2.31464260e+01, -3.51604471e-01, -3.17832656e+00, 1.00372300e+01, -1.61469264e+00, -1.60989737e+01, 1.11154662e+01, 1.09561070e+01, -9.67112079e+00, -1.29745326e+01, -4.65494202e+00, -6.70755219e+00, -1.03939779e+01, 8.37675000e+00, 3.12524698e+01, -1.53137739e+01, -2.06110839e+01, 1.59406804e+01, -4.03199917e+00, -1.49199835e+00, 7.05200045e+00, 1.06014171e+00, -1.31081470e+01, 5.07304872e+00, 2.35326723e+00, -1.39062333e+01, -1.75261092e+00, 8.47353482e+00, -6.98637568e+00, 2.00951915e+00, -1.92542707e+00, -9.72850401e+00, 5.31408831e+00, -1.54752085e+00, -5.14016494e+00, -1.39253815e+00, 2.63586438e+00, -2.71908924e+00, -1.28157534e+01, 5.74664207e+00, -1.28925359e+00, 1.88738375e+00, -4.30273078e+00, -3.24136453e+00, -9.50719529e+00, 6.78074769e+00, -1.36857512e+01, -4.54982811e+00, -1.45929853e+01, 1.56507370e+01, -1.06590613e+01, 4.13557066e+00, 3.61533345e+00, -2.28614188e+00, 1.73247689e+01, -1.61510952e+01, 2.03890276e+00, -1.48451926e+01, 4.68794055e+00, 1.77058883e+01, -2.31952146e+00, 5.23367146e+00, -1.06210304e+01, 1.32771819e+01, -9.12407096e+00, -2.60442685e+00, 1.50278233e+00, -7.17972634e+00, -9.67062311e+00, -8.37286975e-01, 2.31937509e+01, -1.94233697e-01, -7.73553395e+00, 3.77773950e+00, 7.40673674e+00, -2.49166091e+01, 3.19865835e+00, -7.58386005e+00, -1.12040824e+01, 5.83824924e+00, 5.47367141e+00, 3.99207219e+00, 8.58406825e+00, -1.40845479e+01, 1.94802696e+00, -1.01578248e+01, 1.32997962e-01, 1.63139507e+01, -2.46211213e+00, -4.22864395e+00, 8.94992803e+00, -5.13634828e+00, 7.13446506e+00, -2.38514236e+00, 2.54474598e+00, 7.84468576e+00, -9.04492160e+00, -8.86323503e+00, 7.54507654e+00, 1.21215440e+01, 1.27337835e+01, 4.37408655e+00, 1.79538125e+00, -5.57465436e-01, -5.21294199e+00, -1.07583453e+01, 7.40840138e+00, 6.85282455e+00, -1.61884130e+00, 1.63164993e+00, 1.02693998e+00, 4.92771156e+00, 1.29973238e+01, -1.68043303e+01, 1.35002819e+01, 1.31728915e+01, 7.56955107e+00, 7.93402672e-01, -1.11380959e-01, 9.26499313e+00, 1.25700956e+00, -1.01862551e+01, -1.33434017e+01, -2.18214217e+00, -1.89323532e+00, -1.55382459e+01, 5.59983934e-01, -5.22475822e-01, 2.64141961e+00, -4.49060850e+00, 1.01531600e+01, -3.47463318e-01, 1.59469957e+01, 1.94261775e+00, 1.51187866e+01, 3.78539101e+00, -3.64727746e+00, 5.06295908e+00, 3.28383669e+01, -2.70969346e+00, -9.27458391e+00, -1.18565422e+01, 9.65364650e+00, -1.71147435e+01, 6.45984126e+00, 1.82504707e+01, 1.35565677e+01, -3.32956733e-01, -7.68038640e+00, -5.87207851e+00, -2.13271467e+00, 1.92104683e+01, 4.56514784e+00, 2.25312873e+00, 4.20042679e-01, -8.60657622e+00, -8.65308584e+00, 2.86560897e+00, 1.97439116e+01, 3.60155466e-01, -1.07856463e+01, 1.32472433e+00, 9.49806866e+00, -4.61222450e+00, -8.46663007e+00, -6.85382156e+00, 9.22569531e-01, -8.37193202e+00, -6.89565833e+00, -5.50657470e+00, -1.30255652e+01, -1.33108062e+01, -2.65502328e+00, 7.58092714e+00, -7.22479555e-01, 1.56488075e-01, -1.65751327e+00, 1.05258324e+00, -3.34767346e+00, 1.69646687e+00, 9.87078309e+00, 2.30922223e+00, -8.83721454e+00, -2.19407621e+00, -5.82749294e+00, 6.14662323e+00, -1.69011688e+01, -1.14262620e+01, 1.45255052e+01, -8.16873522e+00, 1.13026752e+00, 2.70605179e+01, 1.00910122e+01, 7.24495551e+00, -7.03020719e+00, -2.03717750e+00, -1.46797367e+01, -3.43924883e+00, -6.53415557e+00, 5.89539189e+00, -1.00000828e+01, -4.25333021e+00, -3.52242922e+00, 9.22212008e+00, 2.47828079e+00, -3.28830354e+00, -9.24154738e+00, 2.53782604e+00, 7.12596233e+00, 1.10824317e+01, 4.90718884e+00, -9.55163105e+00, -2.25812348e+01, -1.46110933e+01, 1.00888242e+01, -3.56467151e+00, 1.16325029e+01, -1.13988799e+01, 7.85638077e+00, 3.40747701e+00, 9.74082572e-01, 7.58562710e+00, 7.76051019e+00, -8.79959347e+00, 2.88118272e-01, -2.92936242e+00, 3.00956198e+00, -3.91772387e+00, -4.57747464e-01, -1.32331071e+01, -2.33767481e+00, 2.07534312e+00, -3.57745947e+00, -1.39493386e+00, 8.20378381e+00, 2.13509595e+01, 7.97642561e+00, 8.18724974e+00, 1.93661177e+00, -5.18504439e+00, -5.22291850e+00, 1.18701255e+01, 3.87904397e-01, -1.10183939e+01, 9.83390503e+00, -2.30701667e+00, 3.69891181e+00, -6.02429156e+00, 8.75030694e+00, 7.91525628e+00, 1.18526554e+01, -1.79657139e+01, -3.90350781e+00, 1.50716227e+01, 4.02015946e+00, -5.62113677e+00, 9.46094742e+00, -1.48572373e+00, -1.27032671e+01, -7.03352643e+00, 9.82479951e+00, 1.92152020e+01, 8.49749965e+00, -1.81785598e+01, -3.43534225e+00, 2.59734581e+01, 6.23137355e+00, 7.87794704e+00, -1.45529692e-01, 4.02202448e+00, -6.73081483e+00, 2.14494505e+00, -8.31388127e+00, 1.73312378e+00, 1.56871783e+01, -1.60641060e+01, 3.55238750e+00, 5.95838394e+00, 1.22700820e+00, 4.12515909e+00, -9.95603224e+00, -1.02433907e+01, 1.11689547e+00, 1.02413476e+01, -7.76592943e+00, 1.04368442e+01, 2.26167339e+01, -1.41447677e+01, -1.75778323e+01, -3.45267720e+01, -1.12190756e+01, -1.55673555e+00, -7.91274123e-01, 2.35610011e-01, 2.31504752e+01, 2.32985629e-01, -6.32232562e+00, 1.16417684e+01, -1.67148245e+00, 5.03288546e+00, -1.20637508e+01, -7.02688738e+00, 1.03661032e+00, 3.16270173e+00, -5.03456948e+00, 5.92380327e+00, 9.17797672e+00, -9.11971590e-01, -1.07826115e+01, -9.25162926e+00, -7.36205493e+00, 4.75720384e+00, 1.75789841e+01, 1.26319010e+01, -3.63804226e+00, -1.28715359e+01, -2.56557835e+00, -8.83118810e+00, 5.50802645e+00, 6.94064547e+00, -9.71243980e-01, -1.00583492e+01, -1.17688629e+01, 2.30010523e+00, -9.90486435e+00, -5.34841754e+00, 7.79398128e+00, -8.03009231e+00, -5.91880624e+00, -1.05246517e+01, 3.79001175e+00, 8.71405367e+00, 6.99300477e+00, -3.35569505e+00, 4.86258601e+00, -1.16247990e+00, -8.47643932e+00, -1.21417947e+00, 1.03621090e+01, -8.65716671e+00, 8.21886110e-01, -1.08927501e+01, 8.72344790e+00, 1.82119898e+01, 1.44608308e+00, -1.12468263e+01, -1.67661377e+01, -3.98453337e+00, -2.82250597e+01, 4.38935935e+00, 1.38357135e+00, 2.08443285e+00, -1.72060730e+00, -5.96423361e+00, -4.23793762e+00, 4.07097788e+00, 2.22722146e+01, -1.25517052e+01, -4.86327623e+00, -7.96521580e+00, -6.36944233e+00, -3.44678117e+00, -3.49622034e+01, -9.46776180e+00, 1.58045054e+01, -2.20240498e+00, 1.01166965e+01, 1.50328018e+00, -1.44407725e+01, 2.52839808e+00, -1.50964708e+01, -1.42731332e+01, 1.49675525e+01, 2.15997015e+01, -1.28426471e+00, -5.50841256e-01, -2.50703726e+00, -1.36785429e+01, 1.52443102e+00, 4.36427597e+00, -8.81314563e+00, 6.53112666e+00, -8.24374918e+00, 1.78384732e+01, -2.13786256e+00, -5.08032891e+00, 1.37694152e+01, -1.23055783e+01, -1.58905020e+00, 5.97813602e+00, -6.51755710e+00, -1.28039423e+01, -7.95883554e+00, 1.54381101e+01, 4.19502632e+00, -3.75454903e+00, 1.26285113e+01, 5.30076657e+00, -4.34528876e+00, 1.16977730e+00, 3.08603489e-01, 4.70528704e+00, -6.27100025e+00, -1.12787571e+01, 1.21564711e+01, -8.83741213e-01, -7.59274835e+00, -5.65994452e+00, -1.16515250e+01, 2.41548737e+00, -7.42722573e+00, 8.89376176e+00, -1.51635556e+01, -2.08344246e+01, 1.35352384e+01, -3.49637065e+00, 5.69291678e+00, -3.25227612e+00, 1.22119603e+01, -7.50515902e+00, 1.48719153e+01, 1.01073768e+00, -2.30877365e+00, -8.29665366e+00, 1.74645116e+01, -1.26404700e-01, 3.80787774e+00, 1.00126443e+01, -1.60998268e+01, -9.03005209e+00, 2.95432442e+00, 6.74394005e+00, -7.04557512e+00, 1.75302056e+01, 1.02495955e+01, -2.10805478e+01, -5.07298669e+00, 4.35782104e+00, -9.07833931e+00, -1.33758851e+01, -1.24995697e+01, 7.91468783e+00, -7.13872282e+00, 6.79178951e+00, 6.35064710e+00, -9.11889115e+00, 8.45397813e+00, 1.04585914e+01, -1.57727161e+01, 2.13295084e+00, 9.49088458e-01, -1.32727409e+01, -1.38501742e+01, 4.38288481e-01, 1.83844073e+01, -1.09147680e+01, 2.26977097e+00, 2.45940506e+00, -1.48774685e+01, -1.18313746e+01, -9.82379455e+00, 4.01745963e+00, -7.42273586e+00, 8.01318281e+00, 9.69735024e-01, 2.85630964e+00, 9.38378679e+00, 9.58207559e-01, 7.48772423e+00, -8.79902017e+00, 2.37350450e+00, 4.58325640e+00, 1.19846890e+01, 4.10866433e+00, -4.80410547e+00, -1.15360771e+01, -4.20723254e+00, -6.41557156e+00, 8.30875661e+00, -5.96608864e+00, -5.57751878e+00, -9.44906955e+00, 1.36788548e+01, 7.94298613e-01, -1.21284414e+01, -2.45546643e+01, 4.27025436e-01, 2.84188590e+00, -1.26468621e+00, 9.07265006e-02, 2.78531786e+00, 8.69577958e+00, -3.87924390e+00, -1.26081625e+01, -1.76429525e+01, 3.59339826e+00, -7.32972202e+00, -3.28515711e+00, -2.50806585e+00, 1.13958012e+00, 7.80794036e+00, 8.44887577e+00, -1.22395122e+01, -5.01508391e+00, 1.56196461e+01, 4.26980229e+00, 1.14846152e+01, -1.18073767e+01, -1.23700350e+01, -8.11142965e-01, 1.49241895e+01, 1.22372017e+00, -7.53149505e+00, -4.63471614e+00, 1.72219740e+01, 1.09561422e+01, -4.11058529e+00, 8.89971569e+00, -1.15737223e+01, 3.94879827e+00, 2.18298100e+00, -4.34502917e+00, -1.33466899e+01, 9.09352300e+00, -1.59927844e+01, 3.57747091e+00, 1.35855805e+01, -8.40038677e+00, 1.16328966e+01, 7.15699703e+00, 8.80669245e-01, 1.17275828e+01, 1.47414805e+01, 5.88804362e+00, -1.59150371e+00, -1.06545423e+01, 1.00255423e+01, 1.20663036e+01, 8.06123324e-01, 4.03648357e-01, 7.84096730e+00, -1.74071742e+01, 4.47464301e+00, -1.75722650e+01, 4.47313881e+00, -1.45939680e+01, 1.42801080e+01, -5.53523963e-01, -6.69319448e+00, 4.56002378e+00, -1.97218233e+01, -9.10953434e-01, 6.15091200e+00, -6.42957663e+00, -1.59237863e+01, 2.46388514e+01, 2.12666781e+01, 5.18832989e+00, 9.14610196e+00, 1.12313206e+01, -7.32722092e+00, -5.55963340e+00, 9.15561983e+00, -6.55319367e+00, 9.75822784e+00, -1.14817292e-01, 1.23015930e+01, 1.53273898e+01, -1.76921635e-01, 8.52313054e-01, -1.18677073e+01, 7.08351249e+00, 9.01773365e+00, -2.69884553e+01, -1.42600567e+01, 2.68131361e+00, 2.52000629e+00, 8.07299070e-01, 2.80643474e+00, -1.14866662e+01, 1.38719619e+01, -8.16009622e+00, -6.55621538e+00, -9.09380752e+00, -8.84945158e-01, 1.63686392e+01, -1.39614359e+01, 2.65244394e+00, -4.62334327e+00, -8.74590899e+00, 7.88765799e+00, 8.41631991e+00, -1.17306928e+01, 9.43578910e-02, -9.61257290e+00, -5.14964561e+00, 2.86381011e+00, -1.45458070e+01, -1.13874806e+01, -1.50623411e+01, 7.52425457e+00, -1.90190006e-01, -5.28150052e+00, 2.50954850e+01, 2.51883737e+00, -3.54531706e+00, -1.60484511e+01, 3.79966722e+00, -3.94808280e+00, 3.12683596e+00, 1.35411600e+01, -1.87526423e+01, 1.02476533e+01, 4.67640330e+00, 2.26878411e+00, -2.08960271e+00, 4.52412481e+00, -4.79671072e+00, 4.24796874e+00, -3.24951026e+00, 1.04451510e+01, -1.31702733e+01, 8.12979887e+00, 2.20554896e+00, -6.00533974e+00, -5.77281251e+00, 3.11857545e-02, -5.79801395e+00, 9.29525631e+00, -2.10996381e+00, 3.31108609e+00, -8.61224131e+00, -1.46578559e+01, -1.06918078e+01, 1.14877788e+01, -1.06475245e+01, -5.31494009e+00, 1.09178416e+01, -1.45857500e+01, 1.19075434e+01, -1.59496566e+00, -4.70573498e+00, -4.23451665e+00, -1.43843920e+01, -1.14001039e+01, -1.08505229e+01, 6.86164776e+00, -3.38578352e+00, -1.49090027e+01, 4.73446655e+00, 1.41685489e+01, 1.10038993e+01, -1.07965388e+01, 6.68648368e+00, -1.24485731e+01, -3.14052489e+00, -2.13798334e+01, 2.35602365e+01, -4.62476335e+00, -1.56897545e+01, 1.75962468e+01, -1.17950847e+01, -1.31994305e+00, 1.07849697e+01, -1.91306549e+01, 3.05235213e+01, 8.98523673e+00, 2.10788765e+01, -2.40527560e+00, -2.37363095e+00, 1.41616848e+01, -6.36989783e+00, 2.82109686e+00, 9.95663982e+00, 1.92856526e+00, 8.69675531e+00, -1.09547496e+01, 1.94778906e+00, 6.33226254e+00, -5.57046962e+00, 9.50699281e+00, -9.05158414e+00, -3.75377807e-01, -1.41419913e+00, 2.32173911e+01, -9.87527083e+00, -1.61647237e+00, -7.36777272e+00, 3.16933886e+00, 1.16578387e+00, 1.88686281e+01, -9.83174223e+00, -5.70181910e+00, 3.41244647e+00, -7.34094266e+00, -5.57953138e+00, -1.72188338e+01, 2.34990257e-01, 1.79150176e+01, 4.12078772e+00, -1.70261470e+01, 3.16419453e-02, 2.92897079e+00, -9.09515871e+00, 7.50902461e+00, -5.51802752e+00, 3.96615738e+00, 1.09857084e+01, -9.56458034e+00, 1.94534766e+00, 1.73310425e+01, 2.10842850e+00, 2.43904644e+01, 5.78201170e+00, -1.60215978e+01, -2.51244774e+00, 5.69878976e+00, -2.28895023e+00, -8.90656192e+00, 2.62803104e+00, -1.54730138e+01, 3.32144719e+00, 5.98715131e+00, 1.12651907e+01, 5.33979387e+00, -1.92347183e+00, 3.48896866e+01, -3.06803559e+00, 6.06003788e+00, -2.57887802e+01, -3.75539996e+00, 2.20175470e-02, -6.69320967e+00, -1.34982286e+01, -1.08985826e+01, -1.20726423e+01, 6.34370611e+00, -7.24783314e+00, 1.17432724e+01, 1.13089945e+01, -1.16554212e+00, 1.38580619e+01, -3.71397674e+00, -6.61208902e+00, -1.83942640e+01, -1.49304369e+01, -1.74801850e+01, 1.24270190e+01, -4.04490313e+00, 1.53893697e+01, -1.94797277e+00, -9.97438179e+00, -1.66634454e+01, -2.41922420e+01, -9.59052311e+00, -5.34888016e+00, 1.32201887e+01, 5.63808910e+00, 5.28304303e+00, -3.39994072e+00, -8.22896090e+00, -4.95051637e+00, 1.29705802e+01, 1.10049861e+00, -7.29796935e+00, -1.16137439e+01, 4.67761932e+00, -1.63064405e+01, 1.35495174e+01, 6.57316303e+00, 7.60394103e+00, 8.44539752e+00, -5.50588822e+00, -4.68837521e+00, 3.65693770e+00, -1.20486574e+01, 2.30145013e+01, 7.16973949e+00, 1.59228209e+01, -3.31048317e-01, -7.09085143e-01, -1.72495480e+00, -8.20207059e-01, 1.31560310e+00, -1.94781658e+01, -5.57077142e+00, 1.13277569e+00, 6.04369393e+00, -5.78326282e+00, 1.00355336e+01, -5.75978942e+00, 3.97441294e+00, -5.10644105e+00, 2.29474518e+01, -1.11114824e+01, -1.65217525e+01, 8.89851483e-01, 1.99217165e-01, 6.08311577e+00, 1.23563485e+01, -3.77322283e+00, 1.49293226e+01, -9.89429772e+00, 6.79068667e+00, -4.86927658e-01, -5.07624993e+00, -2.19101541e+00, -6.44909289e+00, 1.21796192e+00, 5.27684159e-01, -2.54629931e+00, -4.69723963e+00, -5.92514137e+00, -2.77220797e+00, 1.06236390e+01, -4.34598315e+00, 1.23199784e+00, 3.92188029e+00]), 'y': array([1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 0, 1, 1, 0, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 0, 0, 1, 0, 1, 0, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 0, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 0, 0, 1, 0, 1, 1, 0, 0, 1, 1, 1, 1, 0, 0, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 0, 0, 1, 1, 1, 0, 0, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 0, 1, 0, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 0, 0, 0, 1, 0, 0, 1, 0, 1, 1, 1, 1, 0, 0, 1, 1, 0, 1, 1, 1, 1, 0, 0, 1, 1, 1, 0, 1, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 0, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 0, 0, 0, 1, 1, 1, 0, 0, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 0, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 0, 0, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 0, 1, 1, 1, 0, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 0, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 0, 1, 0, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 0, 0, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 0, 1, 0, 0, 1, 0, 0, 1, 0, 1, 1, 0, 1, 0, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 1, 0, 0, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 0, 1, 0, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 0, 0, 1, 0, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0, 1, 1, 1, 1, 0, 1, 0, 0, 0, 1, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 0, 1, 0, 1, 1, 1, 0, 0, 0, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 0, 1, 1, 0, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1]), 'pred': array([1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 0, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 0, 1, 1, 0, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 0, 1, 0, 1, 0, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 0, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 0, 1, 0, 0, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 1, 1, 0, 0, 0, 1, 1, 1, 1, 0, 1, 1, 0, 1, 1, 0, 0, 1, 1, 0, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 0, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 0, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 0, 1, 1, 0, 1, 1, 1, 1, 0, 0, 0, 1, 1, 0, 1, 1, 1, 0, 1, 0, 1, 0, 1, 1, 0, 1, 0, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1])}
1 2 3 4 | r_shuf = generate_classification_results(p_pos=0.8, clf=clf, noise_sd=10, shuffle=True) r_shuf['acc'] r_shuf['bacc'] |
np.float64(0.5)
1 2 3 4 | pred_proba = r['clf'].predict_proba(r['X'].reshape(-1, 1))[:,1] pred_df = pd.DataFrame({'X': r['X'], 'y': r['y'], 'pred_proba': pred_proba}) sns.displot(pred_df, x='pred_proba', hue='y', bins=100) plt.xlabel('Predicted probability') |
Text(0.5, 9.444444444444438, 'Predicted probability')
1 2 3 4 5 6 7 8 9 10 | from sklearn.metrics import roc_curve, roc_auc_score fpr, tpr, _ = roc_curve(r['y'], pred_proba) plt.plot(fpr, tpr) plt.xlabel('False positive rate') plt.ylabel('True positive rate') plt.title('ROC curve') plt.plot([0, 1], [0, 1], linestyle='--', color='gray') # fill area under curve plt.fill_between(fpr, tpr, color='lightgray') |
<matplotlib.collections.PolyCollection at 0x17fdcef60>
1 | roc_auc_score(r['y'], pred_proba) |
np.float64(0.8496426855632958)
1 | roc_auc_score(r_shuf['y'], r_shuf['pred']) |
np.float64(0.5)
Assessing regression peformance¶
1 2 3 4 5 6 7 8 9 | def generate_regression_data(n=50, nfeatures=40, noise_sd=1, seed=None): rng = np.random.RandomState(seed) X = rng.normal(size=(n, nfeatures)) beta = rng.normal(size=nfeatures) y = X @ beta + noise_sd * rng.normal(size=n) return(X, y, beta) X, y, beta = generate_regression_data(seed=1) |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | # first to standard linear regression lr = LinearRegression() lr.fit(X, y) y_pred = lr.predict(X) print(f'R2: {r2_score(y, y_pred):.03}') print(f'correlation: {np.corrcoef(y, y_pred)[0, 1]:.03}') # confirm that r2 is the square of the correlation assert np.allclose(r2_score(y, y_pred), np.corrcoef(y, y_pred)[0, 1]**2) print(f'R2 ({r2_score(y, y_pred):.03}) is ~equal to r^2 ({np.corrcoef(y, y_pred)[0, 1]**2:.03})') plt.plot(y, y_pred, 'o') plt.xlabel('True y') plt.ylabel('Predicted y') plt.title('Linear regression - in sample prediction') |
R2: 0.995 correlation: 0.998 R2 (0.995) is ~equal to r^2 (0.995)
Text(0.5, 1.0, 'Linear regression - in sample prediction')
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | clf = LinearRegression kf = KFold(n_splits=8) pred_y = np.zeros_like(y) for split_idx, (train_idx, test_idx) in enumerate(kf.split(X)): train_X = X[train_idx, :] train_y = y[train_idx] test_X = X[test_idx, :] test_y = y[test_idx] # fit the model if clf is None: clf = RidgeCV model = clf() model.fit(train_X, train_y) pred_y[test_idx] = model.predict(test_X) print('R2: ', r2_score(y, pred_y)) print('correlation:', np.corrcoef(y, pred_y)[0, 1]) print(f'R2 ({r2_score(y, pred_y)}) is not equal to r^2 ({np.corrcoef(y, pred_y)[0, 1]**2})') plt.plot(y, pred_y, 'o') plt.xlabel('True y') plt.ylabel('Predicted y') plt.title('Linear regression - out of sample prediction') |
R2: 0.37077968542027107 correlation: 0.7354056951156511 R2 (0.37077968542027107) is not equal to r^2 (0.540821536408534)
Text(0.5, 1.0, 'Linear regression - out of sample prediction')
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | # shuffled data y_shuf = y.copy() np.random.seed(1) np.random.shuffle(y_shuf) lr = LinearRegression() lr.fit(X, y_shuf) y_pred = lr.predict(X) print(f'R2 (in-sample): {r2_score(y_shuf, y_pred):.03}') print(f'correlation (in-sample): {np.corrcoef(y_shuf, y_pred)[0, 1]:.03}') clf = LinearRegression kf = KFold(n_splits=8) pred_y = np.zeros_like(y) for split_idx, (train_idx, test_idx) in enumerate(kf.split(X)): train_X = X[train_idx, :] train_y = y_shuf[train_idx] test_X = X[test_idx, :] test_y = y_shuf[test_idx] # fit the model if clf is None: clf = RidgeCV model = clf() model.fit(train_X, train_y) pred_y[test_idx] = model.predict(test_X) print(f'R2 (out-of-sample): {r2_score(y_shuf, pred_y):.03f}') print(f'correlation (out-of-sample): {np.corrcoef(y_shuf, pred_y)[0, 1]:.03f}') print('') print(f'R2 (mean prediction): {r2_score(y_shuf, np.mean(y_shuf) * np.ones_like(y_shuf)):.03f}') print(f'R2 ({r2_score(y_shuf, pred_y)}) is not equal to r^2 ({np.corrcoef(y_shuf, pred_y)[0, 1]**2})') plt.plot(y, pred_y, 'o') plt.xlabel('True y') plt.ylabel('Predicted y') plt.title('Linear regression (shuffled y)- out of sample prediction') |
R2 (in-sample): 0.889 correlation (in-sample): 0.943 R2 (out-of-sample): -13.970 correlation (out-of-sample): -0.121 R2 (mean prediction): 0.000 R2 (-13.970445357380074) is not equal to r^2 (0.01465717598485716)
Text(0.5, 1.0, 'Linear regression (shuffled y)- out of sample prediction')
1 |