testing_notebook.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "import torch\n",
    "import pyro\n",
    "from pyro.optim import Adam\n",
    "import congas as cg\n",
    "from congas.models import MixtureGaussian,MixtureGaussianEXP, MixtureGaussianOld, MixtureGaussianNorm\n",
    "from pyro.infer import TraceEnum_ELBO\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "outputs": [],
   "source": [
    "data = cg.simulation_data\n",
    "\n",
    "interface = cg.Interface()"
   ],
   "metadata": {
    "collapsed": false,
    "pycharm": {
     "name": "#%%\n"
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "outputs": [],
   "source": [
    "interface.set_model(MixtureGaussian)\n",
    "interface.set_optimizer(Adam)\n",
    "interface.set_loss(TraceEnum_ELBO)\n",
    "interface.initialize_model(data)\n",
    "interface.set_model_params({'K': 2})"
   ],
   "metadata": {
    "collapsed": false,
    "pycharm": {
     "name": "#%%\n"
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Running MixtureGaussian on 1000 cells wiht 32 segments for 100 steps\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "ELBO: 108.532046875  : 100%|██████████| 100/100 [00:02<00:00, 45.58it/s]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "Done!\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "\n"
     ]
    }
   ],
   "source": [
    "ll = interface.run(steps = 100, MAP = False)\n",
    "\n"
   ],
   "metadata": {
    "collapsed": false,
    "pycharm": {
     "name": "#%%\n"
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Computing assignment probabilities\n"
     ]
    },
    {
     "data": {
      "text/plain": "{'param_mixture_weights': array([0.62456715, 0.37543282], dtype=float32),\n 'param_cnv_probs': array([[1.993137  , 1.9512253 , 0.97713286, 2.030496  , 1.9912943 ,\n         1.9611102 , 1.9545438 , 1.0185893 , 1.99126   , 2.047696  ,\n         0.96906763, 1.9631591 , 1.9941092 , 1.9966292 , 2.9202852 ,\n         1.9832568 , 1.9447126 , 2.071764  , 1.9576341 , 3.01876   ,\n         1.0150421 , 2.049042  , 1.987777  , 2.031935  , 1.9752474 ,\n         2.0179265 , 1.9542118 , 2.0840023 , 1.9630741 , 1.9711814 ,\n         0.9981359 , 1.9648991 ],\n        [2.0567508 , 2.0744908 , 1.0469588 , 2.0200458 , 2.1062887 ,\n         2.0600398 , 2.0227973 , 2.0686007 , 2.0288637 , 2.0763376 ,\n         1.0171864 , 0.99372894, 1.9904584 , 1.961207  , 3.011462  ,\n         2.0043912 , 2.0322027 , 2.0405326 , 1.9781868 , 3.0026674 ,\n         0.99886215, 1.925858  , 2.0107944 , 2.0725794 , 1.9756205 ,\n         1.9786181 , 2.0418289 , 2.0876122 , 2.0635183 , 2.0131507 ,\n         1.0011046 , 2.0834    ]], dtype=float32),\n 'param_cnv_var': array([[0.02377643, 0.02274038, 0.02529251, 0.02452719, 0.03262904,\n         0.02407338, 0.01637081, 0.01826015, 0.0289092 , 0.02495312,\n         0.03668812, 0.02763183, 0.01980266, 0.02653421, 0.01931815,\n         0.02371373, 0.02566107, 0.02083877, 0.02397491, 0.02335479,\n         0.03096252, 0.01820603, 0.02565992, 0.02488994, 0.03438226,\n         0.01673838, 0.02223288, 0.0238341 , 0.03557177, 0.02446084,\n         0.0198369 , 0.01789428],\n        [0.01896837, 0.02240983, 0.02952329, 0.0183604 , 0.02798725,\n         0.02034971, 0.02691661, 0.02712283, 0.02615986, 0.02305543,\n         0.03250505, 0.02383401, 0.02742521, 0.02799658, 0.02975408,\n         0.02100918, 0.02733755, 0.02307964, 0.025878  , 0.02269272,\n         0.03175036, 0.0277283 , 0.02935812, 0.02127178, 0.02185271,\n         0.02689291, 0.02910227, 0.0338094 , 0.03406469, 0.0209972 ,\n         0.02289767, 0.02166376]], dtype=float32),\n 'param_norm_factor': array([3.1190257, 2.7992475, 3.5128503, 2.1434107, 3.7289336, 2.960798 ,\n        3.0892174, 4.5261545, 3.7130797, 3.718512 , 3.3710082, 2.8293276,\n        2.364702 , 3.2941933, 2.5859997, 2.8515253, 3.0790987, 4.9343047,\n        3.269153 , 2.61253  , 1.9230741, 3.1690233, 4.5211344, 2.4075902,\n        3.4743335, 5.027904 , 2.546969 , 3.9637408, 4.4059043, 2.8707612,\n        2.2964401, 3.029691 , 4.6749973, 6.447907 , 3.0278642, 2.7032425,\n        3.8907185, 3.676483 , 3.7034767, 2.7180393, 2.9273446, 2.2508237,\n        4.6830354, 2.7477715, 4.0007377, 8.572683 , 2.8583794, 4.3220353,\n        5.405804 , 3.4778101, 3.151938 , 5.547353 , 2.2304802, 3.7490444,\n        2.5938919, 6.754978 , 3.8989737, 4.028042 , 5.483395 , 4.2586656,\n        4.1523833, 3.6333308, 4.95469  , 2.6490686, 2.439523 , 4.929748 ,\n        3.2915547, 4.7448606, 6.5181284, 5.9679594, 3.4483023, 3.1849236,\n        4.257128 , 2.875014 , 1.7196424, 4.933851 , 2.7314632, 4.732689 ,\n        8.875888 , 3.4734123, 3.2043462, 3.5965195, 2.307862 , 6.376502 ,\n        3.8243294, 3.3276205, 3.6778982, 3.8034875, 4.543658 , 4.9693093,\n        2.4101229, 3.508202 , 4.111838 , 5.8469286, 5.8187203, 3.6448898,\n        2.114039 , 4.5750875, 4.325364 , 4.1107087, 4.9274187, 2.9650917,\n        5.150351 , 5.256534 , 6.3771996, 1.9064791, 3.9317172, 3.4708712,\n        3.5277917, 4.1944923, 5.527309 , 3.0099235, 2.4922981, 5.2736573,\n        4.430607 , 2.3919404, 3.436793 , 2.81765  , 5.2784543, 3.3665156,\n        2.7352457, 5.667657 , 4.212317 , 1.8521428, 3.6442537, 3.8856542,\n        4.7537794, 3.115905 , 2.2962713, 4.178445 , 7.705163 , 3.7194364,\n        3.46405  , 2.314091 , 3.9946249, 6.684581 , 2.8856618, 4.423607 ,\n        5.3614693, 6.084858 , 5.372277 , 1.2393357, 1.5423161, 1.8705053,\n        2.2843153, 3.4493704, 2.3663874, 5.4453573, 2.4880743, 3.9431126,\n        4.2333117, 4.871686 , 3.2048464, 2.622251 , 2.186492 , 2.9516451,\n        4.3936234, 4.8256373, 3.7792163, 8.040735 , 4.244954 , 3.2756345,\n        3.3237674, 2.7522001, 1.5696131, 1.9434128, 2.4321995, 4.3157454,\n        3.430007 , 3.423284 , 2.1567967, 4.083838 , 4.080395 , 2.5676334,\n        3.4215286, 2.3942094, 3.587107 , 1.2281287, 3.5764654, 3.182    ,\n        1.8041539, 3.8045037, 2.4511943, 6.007921 , 2.482361 , 3.73877  ,\n        4.3991227, 2.6882305, 5.445129 , 2.6491988, 4.464013 , 2.4811893,\n        3.223545 , 2.398718 , 4.2556453, 2.5394723, 3.8574822, 1.7609324,\n        5.023203 , 5.550514 , 2.6769881, 2.7155068, 4.8228917, 2.7611141,\n        3.1435926, 2.8887033, 3.2572074, 1.913541 , 3.7815583, 5.2045946,\n        2.7649581, 8.0558195, 2.3619602, 2.7384903, 3.7467806, 4.4733887,\n        3.4514384, 4.6008325, 6.875366 , 3.9495058, 2.2824345, 1.5811357,\n        4.5116086, 2.5261965, 3.0548968, 4.399402 , 2.0936306, 3.2922103,\n        3.7864943, 3.8228557, 5.477206 , 6.7205753, 3.081373 , 4.1011715,\n        8.099746 , 4.05384  , 4.987118 , 4.534574 , 3.4006345, 2.6541846,\n        2.2439134, 2.140879 , 8.547835 , 4.5390515, 3.0071547, 2.5903356,\n        2.7102273, 4.016413 , 2.211813 , 2.9699304, 3.631533 , 4.7568474,\n        4.754161 , 3.453219 , 2.2500088, 2.84244  , 5.2036977, 2.3033538,\n        2.4263625, 5.259865 , 1.5079795, 5.7941275, 4.4207516, 2.1296015,\n        2.9957597, 3.9420743, 3.462792 , 1.0662404, 5.859697 , 2.29586  ,\n        3.9525886, 3.0863445, 4.1003666, 3.7146354, 2.3251605, 5.138755 ,\n        3.6017587, 3.835959 , 2.67674  , 9.126899 , 3.7228677, 3.3933842,\n        3.9023495, 1.9565324, 2.6311073, 1.9112092, 4.871694 , 2.614499 ,\n        1.9727334, 2.3847892, 5.541305 , 3.5354261, 3.3122652, 2.8452334,\n        3.699136 , 2.9344127, 3.1936636, 4.651107 , 5.0535326, 4.586577 ,\n        5.045077 , 3.3274877, 3.8142226, 3.1257057, 6.329612 , 6.0108976,\n        2.985883 , 4.1723046, 1.0636976, 3.160011 , 4.000482 , 2.7809184,\n        5.5317388, 3.3240798, 4.6556644, 3.088372 , 4.9175534, 5.0350914,\n        2.734081 , 5.7772284, 2.6648993, 1.3531598, 4.511877 , 3.4447503,\n        3.1567707, 4.1221614, 8.786886 , 2.5583653, 2.6652904, 2.0867856,\n        2.358185 , 3.6258588, 2.3395462, 1.4257873, 4.5662775, 3.8656218,\n        5.096338 , 3.5268939, 5.310997 , 6.272574 , 3.6191986, 1.8042551,\n        3.4096158, 4.7864842, 2.7816367, 3.0452728, 3.694577 , 4.439727 ,\n        5.607855 , 6.4040136, 6.739098 , 5.5226502, 3.7705636, 5.0557833,\n        1.9363831, 1.8976271, 2.6145668, 2.9159663, 4.6359496, 4.408431 ,\n        4.513182 , 6.8606396, 1.7450494, 4.209497 , 3.4941044, 4.551271 ,\n        4.064239 , 4.1256957, 4.6496925, 4.0671754, 4.435733 , 2.667894 ,\n        4.260536 , 3.9779544, 3.8129652, 1.3346593, 4.6318192, 5.6703467,\n        3.0708349, 3.6919   , 3.1088893, 4.6393733, 6.245425 , 3.8820765,\n        3.9977715, 5.7083964, 3.4891715, 7.147085 , 3.201178 , 4.5209074,\n        2.8444328, 3.644837 , 4.3611193, 3.4719145, 4.0850224, 3.3322115,\n        4.474007 , 2.5603206, 3.829772 , 1.9546607, 3.222748 , 3.1806452,\n        5.8310914, 3.9380877, 2.426302 , 4.0546536, 3.112179 , 3.2367427,\n        3.2581084, 4.1723886, 3.3369584, 3.3311827, 4.2080445, 1.8882502,\n        4.2596517, 3.794043 , 3.6670852, 3.19413  , 2.3640788, 4.6380696,\n        3.550695 , 6.203058 , 4.248448 , 3.0800264, 6.7663836, 2.264793 ,\n        2.6692967, 4.086229 , 5.878418 , 1.7770808, 5.678126 , 5.005485 ,\n        3.5872033, 7.6207423, 4.190727 , 3.4940064, 6.561843 , 5.3998075,\n        6.9355636, 3.2650979, 5.2435184, 7.1398053, 4.4015536, 2.972782 ,\n        2.8628616, 1.4985827, 4.074157 , 0.6240093, 5.348677 , 4.390218 ,\n        2.460052 , 4.8284335, 6.630663 , 3.0863936, 3.4495218, 3.6771936,\n        5.3111863, 3.3944273, 5.0555015, 4.154667 , 3.255484 , 2.4347086,\n        4.1670113, 1.5970999, 4.6569085, 2.746981 , 2.1256785, 3.1984386,\n        3.626732 , 3.1553228, 3.0118577, 3.5915453, 1.9975641, 3.920635 ,\n        2.9025474, 7.5115337, 3.1850965, 5.3875756, 3.221542 , 3.426287 ,\n        2.3921735, 2.8521252, 8.385745 , 3.7680798, 2.348336 , 3.2566195,\n        3.8284445, 3.836702 , 3.0803387, 3.0002937, 6.7117243, 4.0994062,\n        4.662666 , 2.9407287, 4.815272 , 5.825652 , 3.012925 , 2.608777 ,\n        2.8908641, 2.3730953, 2.7260914, 1.6904454, 2.5669417, 2.036139 ,\n        4.7774158, 3.1961312, 0.7447056, 5.183411 , 4.7347007, 3.03432  ,\n        4.915232 , 3.8295195, 4.8283606, 4.7509394, 4.355939 , 4.3722863,\n        4.6679544, 6.4249163, 2.5882137, 2.1543868, 4.363667 , 4.3510685,\n        2.635193 , 1.2964159, 2.9159393, 2.405541 , 5.12688  , 3.6647005,\n        1.4897107, 5.093025 , 3.6358283, 4.264708 , 3.6431823, 4.686939 ,\n        2.6858547, 6.4839153, 2.3026023, 4.7794714, 1.6492416, 3.410083 ,\n        3.0631733, 4.936561 , 2.7762995, 3.7835257, 3.912782 , 2.0816894,\n        5.582066 , 3.8739216, 2.6742299, 3.0656765, 2.5827255, 3.9767566,\n        2.8333185, 3.4676917, 5.657082 , 6.747755 , 4.5037575, 5.5843887,\n        6.9151664, 2.5353386, 3.7609363, 2.7356977, 6.4911203, 4.0637355,\n        2.793302 , 3.138492 , 5.873762 , 3.9412174, 6.980996 , 5.283818 ,\n        4.736816 , 3.703478 , 3.3369024, 3.6716073, 2.7497122, 2.8740838,\n        3.900586 , 3.8696744, 2.2895489, 3.8742878, 4.3364687, 5.5529504,\n        2.353607 , 6.474771 , 4.142142 , 2.387467 , 4.111779 , 2.4781783,\n        2.6517532, 4.893108 , 2.453495 , 4.2260222, 2.9744577, 5.7283444,\n        3.8301258, 4.231448 , 2.2941718, 3.60802  , 2.0906596, 5.8420625,\n        2.7307758, 6.3066382, 6.7055435, 1.5423368, 5.6657276, 3.2485244,\n        2.8882601, 4.004339 , 1.847643 , 1.4120632, 3.2407255, 4.1909   ,\n        8.404065 , 4.8189363, 5.031163 , 5.9650855, 2.3413231, 5.003532 ,\n        3.2661264, 5.5571756, 2.405921 , 5.839503 , 3.0005283, 4.7379484,\n        3.8947046, 3.624711 , 3.4215245, 4.673645 , 1.8903553, 2.1047478,\n        3.2372446, 5.820283 , 3.8035583, 3.3502638, 2.4827783, 1.7268649,\n        3.3759305, 5.3800845, 3.5129266, 1.4369999, 2.2164087, 4.5919857,\n        2.1575341, 4.0667844, 4.053088 , 2.446992 , 2.3072622, 3.789044 ,\n        5.3062277, 2.908812 , 5.9825377, 2.4350088, 3.7714653, 3.3351939,\n        4.5932527, 5.2673817, 3.0639362, 2.6285853, 3.3249793, 6.0165324,\n        3.2462385, 3.4877713, 3.6034796, 5.8592277, 2.5007834, 2.614583 ,\n        3.1867154, 4.882664 , 4.210428 , 4.4365463, 4.867529 , 4.759643 ,\n        4.508751 , 6.1148243, 1.498967 , 4.7677245, 3.1501045, 5.2048635,\n        4.102511 , 3.3191445, 5.8331027, 1.9316542, 5.5704975, 4.6798673,\n        3.9871821, 3.1913507, 3.055228 , 4.37613  , 6.5798097, 4.5361357,\n        5.456791 , 3.6505742, 4.6965685, 1.6926435, 2.6132069, 3.9850543,\n        2.6601455, 3.0839026, 2.7770686, 2.5514014, 3.2019594, 2.8403447,\n        3.4608974, 2.2142584, 3.8882184, 2.9346461, 3.1258776, 4.6488767,\n        3.6133358, 3.6530855, 3.5300014, 2.7996993, 2.2734597, 3.3285615,\n        2.5872633, 2.9139547, 3.095898 , 4.773505 , 3.249019 , 2.584856 ,\n        2.0254254, 3.17236  , 4.7715573, 2.4083219, 3.5404522, 5.1041136,\n        2.6736   , 3.788666 , 4.4570413, 2.9727957, 2.6383295, 3.0334694,\n        4.7543283, 6.261961 , 3.1225476, 2.493885 , 3.973483 , 3.4935927,\n        3.5881078, 2.6214375, 3.142095 , 2.2299855, 4.532569 , 2.6218643,\n        3.8482232, 8.726705 , 2.9802709, 4.468245 , 5.285523 , 3.3456676,\n        3.2462764, 5.4517107, 2.2823668, 3.877701 , 2.6673255, 6.709724 ,\n        3.84769  , 3.9958222, 5.4008074, 4.325981 , 4.009845 , 3.6046233,\n        4.840132 , 2.6899116, 2.5243802, 4.808377 , 3.267415 , 4.610862 ,\n        6.223365 , 5.6585517, 3.4260855, 3.103339 , 4.2847476, 2.9090574,\n        1.6470392, 4.940646 , 2.690134 , 4.8323216, 9.155419 , 3.5074215,\n        3.3346176, 3.5599039, 2.2501793, 6.247791 , 3.7888005, 3.203343 ,\n        3.5238752, 3.7115057, 4.374646 , 4.9222283, 2.483245 , 3.4244568,\n        3.9477844, 5.8645053, 5.6663175, 3.5873644, 1.8951367, 4.5495505,\n        4.355371 , 4.076623 , 4.5977383, 3.1133955, 5.17691  , 5.302071 ,\n        6.3288493, 1.9452044, 4.0734177, 3.5025437, 3.472053 , 4.0642605,\n        5.445368 , 2.8424025, 2.446443 , 4.936491 , 4.4392343, 2.2434726,\n        3.2578363, 2.7658644, 5.035549 , 3.5015473, 2.8152108, 5.807342 ,\n        4.0494976, 1.7334507, 3.5078602, 4.04418  , 4.534914 , 2.9415023,\n        2.3279026, 4.1934414, 7.770937 , 3.9148793, 3.4858027, 2.3782985,\n        3.689524 , 6.6959434, 2.8923755, 4.56367  , 5.518324 , 6.1890135,\n        5.3927565, 1.3187373, 1.494544 , 1.7178541, 2.3127058, 3.400066 ,\n        2.1801662, 5.554757 , 2.587843 , 3.815029 , 4.217317 , 4.7391596,\n        3.1185226, 2.6807325, 2.215915 , 2.715908 , 4.398035 , 4.8635406,\n        3.6547656, 8.089397 , 4.0214777, 3.166227 , 3.2897577, 2.7307172,\n        1.435074 , 1.8841501, 2.5721076, 4.174543 , 3.6066387, 3.3378525,\n        2.2436097, 3.8694613, 4.063578 , 2.441306 , 3.332861 , 2.237053 ,\n        3.537608 , 1.1322134, 3.5371132, 3.0711484, 1.776819 , 3.708    ,\n        2.611198 , 5.7986856, 2.3461483, 3.7359357, 4.3026733, 2.701383 ,\n        5.3175244, 2.7829843, 4.361685 , 2.1931   , 3.340146 , 2.3253324,\n        4.4203286, 2.68325  , 3.5447197, 1.9904602, 4.955637 , 5.535318 ,\n        2.5562835, 2.7489467, 4.7510867, 2.6153255, 3.109245 , 2.7988698,\n        3.090617 , 1.8635366, 3.9269645, 5.199528 , 2.7154787, 7.934073 ,\n        2.2849827, 2.645242 , 3.5781472, 4.66378  , 3.371467 , 4.6333942,\n        6.926371 , 3.9394033, 2.1754587, 1.5828078, 4.4522004, 2.5768788,\n        3.2353742, 4.2140784, 2.0336819, 3.3501241, 3.5840473, 4.0664926,\n        5.392799 , 6.5245967, 3.0382328, 4.1882377, 8.010257 , 4.0117664,\n        4.9083314, 4.4629703, 3.5256963, 2.6715796, 2.1276312, 2.1889873,\n        8.637401 , 4.5280895, 3.2155988, 2.5897245, 2.576526 , 3.8858867,\n        2.0656958, 3.1019287, 3.8276565, 4.7452226, 4.5764523, 3.5131812,\n        2.2875097, 2.8946154, 4.821066 , 2.374066 , 2.323786 , 4.997047 ,\n        1.5622871, 5.8373065, 4.150182 , 2.2003958, 2.6930213, 3.8780022,\n        3.5904593, 1.0025523, 5.754885 , 2.1295388, 4.0968018, 3.0919256,\n        4.001446 , 3.8791635, 2.1209323, 4.9939885, 3.9198256, 3.885469 ,\n        2.7316425, 9.097986 , 3.918215 , 3.2660062, 3.8415082, 2.0861013,\n        2.4923022, 1.9724542, 4.982806 , 2.674518 , 1.8863924, 2.3168175,\n        5.5113173, 3.6832042, 3.439152 , 2.7786438, 3.6796045, 3.0731583,\n        3.1113293, 4.467496 , 4.8631153, 4.497037 ], dtype=float32),\n 'cnv_probs': array([[1.993137  , 1.9512253 , 0.97713286, 2.030496  , 1.9912943 ,\n         1.9611102 , 1.9545438 , 1.0185893 , 1.99126   , 2.047696  ,\n         0.96906763, 1.9631591 , 1.9941092 , 1.9966292 , 2.9202852 ,\n         1.9832568 , 1.9447126 , 2.071764  , 1.9576341 , 3.01876   ,\n         1.0150421 , 2.049042  , 1.987777  , 2.031935  , 1.9752474 ,\n         2.0179265 , 1.9542118 , 2.0840023 , 1.9630741 , 1.9711814 ,\n         0.9981359 , 1.9648991 ],\n        [2.0567508 , 2.0744908 , 1.0469588 , 2.0200458 , 2.1062887 ,\n         2.0600398 , 2.0227973 , 2.0686007 , 2.0288637 , 2.0763376 ,\n         1.0171864 , 0.99372894, 1.9904584 , 1.961207  , 3.011462  ,\n         2.0043912 , 2.0322027 , 2.0405326 , 1.9781868 , 3.0026674 ,\n         0.99886215, 1.925858  , 2.0107944 , 2.0725794 , 1.9756205 ,\n         1.9786181 , 2.0418289 , 2.0876122 , 2.0635183 , 2.0131507 ,\n         1.0011046 , 2.0834    ]], dtype=float32),\n 'norm_factor': array([3.1190257, 2.7992475, 3.5128503, 2.1434107, 3.7289336, 2.960798 ,\n        3.0892174, 4.5261545, 3.7130797, 3.718512 , 3.3710082, 2.8293276,\n        2.364702 , 3.2941933, 2.5859997, 2.8515253, 3.0790987, 4.9343047,\n        3.269153 , 2.61253  , 1.9230741, 3.1690233, 4.5211344, 2.4075902,\n        3.4743335, 5.027904 , 2.546969 , 3.9637408, 4.4059043, 2.8707612,\n        2.2964401, 3.029691 , 4.6749973, 6.447907 , 3.0278642, 2.7032425,\n        3.8907185, 3.676483 , 3.7034767, 2.7180393, 2.9273446, 2.2508237,\n        4.6830354, 2.7477715, 4.0007377, 8.572683 , 2.8583794, 4.3220353,\n        5.405804 , 3.4778101, 3.151938 , 5.547353 , 2.2304802, 3.7490444,\n        2.5938919, 6.754978 , 3.8989737, 4.028042 , 5.483395 , 4.2586656,\n        4.1523833, 3.6333308, 4.95469  , 2.6490686, 2.439523 , 4.929748 ,\n        3.2915547, 4.7448606, 6.5181284, 5.9679594, 3.4483023, 3.1849236,\n        4.257128 , 2.875014 , 1.7196424, 4.933851 , 2.7314632, 4.732689 ,\n        8.875888 , 3.4734123, 3.2043462, 3.5965195, 2.307862 , 6.376502 ,\n        3.8243294, 3.3276205, 3.6778982, 3.8034875, 4.543658 , 4.9693093,\n        2.4101229, 3.508202 , 4.111838 , 5.8469286, 5.8187203, 3.6448898,\n        2.114039 , 4.5750875, 4.325364 , 4.1107087, 4.9274187, 2.9650917,\n        5.150351 , 5.256534 , 6.3771996, 1.9064791, 3.9317172, 3.4708712,\n        3.5277917, 4.1944923, 5.527309 , 3.0099235, 2.4922981, 5.2736573,\n        4.430607 , 2.3919404, 3.436793 , 2.81765  , 5.2784543, 3.3665156,\n        2.7352457, 5.667657 , 4.212317 , 1.8521428, 3.6442537, 3.8856542,\n        4.7537794, 3.115905 , 2.2962713, 4.178445 , 7.705163 , 3.7194364,\n        3.46405  , 2.314091 , 3.9946249, 6.684581 , 2.8856618, 4.423607 ,\n        5.3614693, 6.084858 , 5.372277 , 1.2393357, 1.5423161, 1.8705053,\n        2.2843153, 3.4493704, 2.3663874, 5.4453573, 2.4880743, 3.9431126,\n        4.2333117, 4.871686 , 3.2048464, 2.622251 , 2.186492 , 2.9516451,\n        4.3936234, 4.8256373, 3.7792163, 8.040735 , 4.244954 , 3.2756345,\n        3.3237674, 2.7522001, 1.5696131, 1.9434128, 2.4321995, 4.3157454,\n        3.430007 , 3.423284 , 2.1567967, 4.083838 , 4.080395 , 2.5676334,\n        3.4215286, 2.3942094, 3.587107 , 1.2281287, 3.5764654, 3.182    ,\n        1.8041539, 3.8045037, 2.4511943, 6.007921 , 2.482361 , 3.73877  ,\n        4.3991227, 2.6882305, 5.445129 , 2.6491988, 4.464013 , 2.4811893,\n        3.223545 , 2.398718 , 4.2556453, 2.5394723, 3.8574822, 1.7609324,\n        5.023203 , 5.550514 , 2.6769881, 2.7155068, 4.8228917, 2.7611141,\n        3.1435926, 2.8887033, 3.2572074, 1.913541 , 3.7815583, 5.2045946,\n        2.7649581, 8.0558195, 2.3619602, 2.7384903, 3.7467806, 4.4733887,\n        3.4514384, 4.6008325, 6.875366 , 3.9495058, 2.2824345, 1.5811357,\n        4.5116086, 2.5261965, 3.0548968, 4.399402 , 2.0936306, 3.2922103,\n        3.7864943, 3.8228557, 5.477206 , 6.7205753, 3.081373 , 4.1011715,\n        8.099746 , 4.05384  , 4.987118 , 4.534574 , 3.4006345, 2.6541846,\n        2.2439134, 2.140879 , 8.547835 , 4.5390515, 3.0071547, 2.5903356,\n        2.7102273, 4.016413 , 2.211813 , 2.9699304, 3.631533 , 4.7568474,\n        4.754161 , 3.453219 , 2.2500088, 2.84244  , 5.2036977, 2.3033538,\n        2.4263625, 5.259865 , 1.5079795, 5.7941275, 4.4207516, 2.1296015,\n        2.9957597, 3.9420743, 3.462792 , 1.0662404, 5.859697 , 2.29586  ,\n        3.9525886, 3.0863445, 4.1003666, 3.7146354, 2.3251605, 5.138755 ,\n        3.6017587, 3.835959 , 2.67674  , 9.126899 , 3.7228677, 3.3933842,\n        3.9023495, 1.9565324, 2.6311073, 1.9112092, 4.871694 , 2.614499 ,\n        1.9727334, 2.3847892, 5.541305 , 3.5354261, 3.3122652, 2.8452334,\n        3.699136 , 2.9344127, 3.1936636, 4.651107 , 5.0535326, 4.586577 ,\n        5.045077 , 3.3274877, 3.8142226, 3.1257057, 6.329612 , 6.0108976,\n        2.985883 , 4.1723046, 1.0636976, 3.160011 , 4.000482 , 2.7809184,\n        5.5317388, 3.3240798, 4.6556644, 3.088372 , 4.9175534, 5.0350914,\n        2.734081 , 5.7772284, 2.6648993, 1.3531598, 4.511877 , 3.4447503,\n        3.1567707, 4.1221614, 8.786886 , 2.5583653, 2.6652904, 2.0867856,\n        2.358185 , 3.6258588, 2.3395462, 1.4257873, 4.5662775, 3.8656218,\n        5.096338 , 3.5268939, 5.310997 , 6.272574 , 3.6191986, 1.8042551,\n        3.4096158, 4.7864842, 2.7816367, 3.0452728, 3.694577 , 4.439727 ,\n        5.607855 , 6.4040136, 6.739098 , 5.5226502, 3.7705636, 5.0557833,\n        1.9363831, 1.8976271, 2.6145668, 2.9159663, 4.6359496, 4.408431 ,\n        4.513182 , 6.8606396, 1.7450494, 4.209497 , 3.4941044, 4.551271 ,\n        4.064239 , 4.1256957, 4.6496925, 4.0671754, 4.435733 , 2.667894 ,\n        4.260536 , 3.9779544, 3.8129652, 1.3346593, 4.6318192, 5.6703467,\n        3.0708349, 3.6919   , 3.1088893, 4.6393733, 6.245425 , 3.8820765,\n        3.9977715, 5.7083964, 3.4891715, 7.147085 , 3.201178 , 4.5209074,\n        2.8444328, 3.644837 , 4.3611193, 3.4719145, 4.0850224, 3.3322115,\n        4.474007 , 2.5603206, 3.829772 , 1.9546607, 3.222748 , 3.1806452,\n        5.8310914, 3.9380877, 2.426302 , 4.0546536, 3.112179 , 3.2367427,\n        3.2581084, 4.1723886, 3.3369584, 3.3311827, 4.2080445, 1.8882502,\n        4.2596517, 3.794043 , 3.6670852, 3.19413  , 2.3640788, 4.6380696,\n        3.550695 , 6.203058 , 4.248448 , 3.0800264, 6.7663836, 2.264793 ,\n        2.6692967, 4.086229 , 5.878418 , 1.7770808, 5.678126 , 5.005485 ,\n        3.5872033, 7.6207423, 4.190727 , 3.4940064, 6.561843 , 5.3998075,\n        6.9355636, 3.2650979, 5.2435184, 7.1398053, 4.4015536, 2.972782 ,\n        2.8628616, 1.4985827, 4.074157 , 0.6240093, 5.348677 , 4.390218 ,\n        2.460052 , 4.8284335, 6.630663 , 3.0863936, 3.4495218, 3.6771936,\n        5.3111863, 3.3944273, 5.0555015, 4.154667 , 3.255484 , 2.4347086,\n        4.1670113, 1.5970999, 4.6569085, 2.746981 , 2.1256785, 3.1984386,\n        3.626732 , 3.1553228, 3.0118577, 3.5915453, 1.9975641, 3.920635 ,\n        2.9025474, 7.5115337, 3.1850965, 5.3875756, 3.221542 , 3.426287 ,\n        2.3921735, 2.8521252, 8.385745 , 3.7680798, 2.348336 , 3.2566195,\n        3.8284445, 3.836702 , 3.0803387, 3.0002937, 6.7117243, 4.0994062,\n        4.662666 , 2.9407287, 4.815272 , 5.825652 , 3.012925 , 2.608777 ,\n        2.8908641, 2.3730953, 2.7260914, 1.6904454, 2.5669417, 2.036139 ,\n        4.7774158, 3.1961312, 0.7447056, 5.183411 , 4.7347007, 3.03432  ,\n        4.915232 , 3.8295195, 4.8283606, 4.7509394, 4.355939 , 4.3722863,\n        4.6679544, 6.4249163, 2.5882137, 2.1543868, 4.363667 , 4.3510685,\n        2.635193 , 1.2964159, 2.9159393, 2.405541 , 5.12688  , 3.6647005,\n        1.4897107, 5.093025 , 3.6358283, 4.264708 , 3.6431823, 4.686939 ,\n        2.6858547, 6.4839153, 2.3026023, 4.7794714, 1.6492416, 3.410083 ,\n        3.0631733, 4.936561 , 2.7762995, 3.7835257, 3.912782 , 2.0816894,\n        5.582066 , 3.8739216, 2.6742299, 3.0656765, 2.5827255, 3.9767566,\n        2.8333185, 3.4676917, 5.657082 , 6.747755 , 4.5037575, 5.5843887,\n        6.9151664, 2.5353386, 3.7609363, 2.7356977, 6.4911203, 4.0637355,\n        2.793302 , 3.138492 , 5.873762 , 3.9412174, 6.980996 , 5.283818 ,\n        4.736816 , 3.703478 , 3.3369024, 3.6716073, 2.7497122, 2.8740838,\n        3.900586 , 3.8696744, 2.2895489, 3.8742878, 4.3364687, 5.5529504,\n        2.353607 , 6.474771 , 4.142142 , 2.387467 , 4.111779 , 2.4781783,\n        2.6517532, 4.893108 , 2.453495 , 4.2260222, 2.9744577, 5.7283444,\n        3.8301258, 4.231448 , 2.2941718, 3.60802  , 2.0906596, 5.8420625,\n        2.7307758, 6.3066382, 6.7055435, 1.5423368, 5.6657276, 3.2485244,\n        2.8882601, 4.004339 , 1.847643 , 1.4120632, 3.2407255, 4.1909   ,\n        8.404065 , 4.8189363, 5.031163 , 5.9650855, 2.3413231, 5.003532 ,\n        3.2661264, 5.5571756, 2.405921 , 5.839503 , 3.0005283, 4.7379484,\n        3.8947046, 3.624711 , 3.4215245, 4.673645 , 1.8903553, 2.1047478,\n        3.2372446, 5.820283 , 3.8035583, 3.3502638, 2.4827783, 1.7268649,\n        3.3759305, 5.3800845, 3.5129266, 1.4369999, 2.2164087, 4.5919857,\n        2.1575341, 4.0667844, 4.053088 , 2.446992 , 2.3072622, 3.789044 ,\n        5.3062277, 2.908812 , 5.9825377, 2.4350088, 3.7714653, 3.3351939,\n        4.5932527, 5.2673817, 3.0639362, 2.6285853, 3.3249793, 6.0165324,\n        3.2462385, 3.4877713, 3.6034796, 5.8592277, 2.5007834, 2.614583 ,\n        3.1867154, 4.882664 , 4.210428 , 4.4365463, 4.867529 , 4.759643 ,\n        4.508751 , 6.1148243, 1.498967 , 4.7677245, 3.1501045, 5.2048635,\n        4.102511 , 3.3191445, 5.8331027, 1.9316542, 5.5704975, 4.6798673,\n        3.9871821, 3.1913507, 3.055228 , 4.37613  , 6.5798097, 4.5361357,\n        5.456791 , 3.6505742, 4.6965685, 1.6926435, 2.6132069, 3.9850543,\n        2.6601455, 3.0839026, 2.7770686, 2.5514014, 3.2019594, 2.8403447,\n        3.4608974, 2.2142584, 3.8882184, 2.9346461, 3.1258776, 4.6488767,\n        3.6133358, 3.6530855, 3.5300014, 2.7996993, 2.2734597, 3.3285615,\n        2.5872633, 2.9139547, 3.095898 , 4.773505 , 3.249019 , 2.584856 ,\n        2.0254254, 3.17236  , 4.7715573, 2.4083219, 3.5404522, 5.1041136,\n        2.6736   , 3.788666 , 4.4570413, 2.9727957, 2.6383295, 3.0334694,\n        4.7543283, 6.261961 , 3.1225476, 2.493885 , 3.973483 , 3.4935927,\n        3.5881078, 2.6214375, 3.142095 , 2.2299855, 4.532569 , 2.6218643,\n        3.8482232, 8.726705 , 2.9802709, 4.468245 , 5.285523 , 3.3456676,\n        3.2462764, 5.4517107, 2.2823668, 3.877701 , 2.6673255, 6.709724 ,\n        3.84769  , 3.9958222, 5.4008074, 4.325981 , 4.009845 , 3.6046233,\n        4.840132 , 2.6899116, 2.5243802, 4.808377 , 3.267415 , 4.610862 ,\n        6.223365 , 5.6585517, 3.4260855, 3.103339 , 4.2847476, 2.9090574,\n        1.6470392, 4.940646 , 2.690134 , 4.8323216, 9.155419 , 3.5074215,\n        3.3346176, 3.5599039, 2.2501793, 6.247791 , 3.7888005, 3.203343 ,\n        3.5238752, 3.7115057, 4.374646 , 4.9222283, 2.483245 , 3.4244568,\n        3.9477844, 5.8645053, 5.6663175, 3.5873644, 1.8951367, 4.5495505,\n        4.355371 , 4.076623 , 4.5977383, 3.1133955, 5.17691  , 5.302071 ,\n        6.3288493, 1.9452044, 4.0734177, 3.5025437, 3.472053 , 4.0642605,\n        5.445368 , 2.8424025, 2.446443 , 4.936491 , 4.4392343, 2.2434726,\n        3.2578363, 2.7658644, 5.035549 , 3.5015473, 2.8152108, 5.807342 ,\n        4.0494976, 1.7334507, 3.5078602, 4.04418  , 4.534914 , 2.9415023,\n        2.3279026, 4.1934414, 7.770937 , 3.9148793, 3.4858027, 2.3782985,\n        3.689524 , 6.6959434, 2.8923755, 4.56367  , 5.518324 , 6.1890135,\n        5.3927565, 1.3187373, 1.494544 , 1.7178541, 2.3127058, 3.400066 ,\n        2.1801662, 5.554757 , 2.587843 , 3.815029 , 4.217317 , 4.7391596,\n        3.1185226, 2.6807325, 2.215915 , 2.715908 , 4.398035 , 4.8635406,\n        3.6547656, 8.089397 , 4.0214777, 3.166227 , 3.2897577, 2.7307172,\n        1.435074 , 1.8841501, 2.5721076, 4.174543 , 3.6066387, 3.3378525,\n        2.2436097, 3.8694613, 4.063578 , 2.441306 , 3.332861 , 2.237053 ,\n        3.537608 , 1.1322134, 3.5371132, 3.0711484, 1.776819 , 3.708    ,\n        2.611198 , 5.7986856, 2.3461483, 3.7359357, 4.3026733, 2.701383 ,\n        5.3175244, 2.7829843, 4.361685 , 2.1931   , 3.340146 , 2.3253324,\n        4.4203286, 2.68325  , 3.5447197, 1.9904602, 4.955637 , 5.535318 ,\n        2.5562835, 2.7489467, 4.7510867, 2.6153255, 3.109245 , 2.7988698,\n        3.090617 , 1.8635366, 3.9269645, 5.199528 , 2.7154787, 7.934073 ,\n        2.2849827, 2.645242 , 3.5781472, 4.66378  , 3.371467 , 4.6333942,\n        6.926371 , 3.9394033, 2.1754587, 1.5828078, 4.4522004, 2.5768788,\n        3.2353742, 4.2140784, 2.0336819, 3.3501241, 3.5840473, 4.0664926,\n        5.392799 , 6.5245967, 3.0382328, 4.1882377, 8.010257 , 4.0117664,\n        4.9083314, 4.4629703, 3.5256963, 2.6715796, 2.1276312, 2.1889873,\n        8.637401 , 4.5280895, 3.2155988, 2.5897245, 2.576526 , 3.8858867,\n        2.0656958, 3.1019287, 3.8276565, 4.7452226, 4.5764523, 3.5131812,\n        2.2875097, 2.8946154, 4.821066 , 2.374066 , 2.323786 , 4.997047 ,\n        1.5622871, 5.8373065, 4.150182 , 2.2003958, 2.6930213, 3.8780022,\n        3.5904593, 1.0025523, 5.754885 , 2.1295388, 4.0968018, 3.0919256,\n        4.001446 , 3.8791635, 2.1209323, 4.9939885, 3.9198256, 3.885469 ,\n        2.7316425, 9.097986 , 3.918215 , 3.2660062, 3.8415082, 2.0861013,\n        2.4923022, 1.9724542, 4.982806 , 2.674518 , 1.8863924, 2.3168175,\n        5.5113173, 3.6832042, 3.439152 , 2.7786438, 3.6796045, 3.0731583,\n        3.1113293, 4.467496 , 4.8631153, 4.497037 ], dtype=float32),\n 'mixture_weights': array([0.62456715, 0.37543282], dtype=float32),\n 'assignment_probs': array([[7.9461239e-02, 1.0000000e+00, 1.0000000e+00, ..., 2.2426982e-12,\n         3.7202748e-14, 6.0761278e-14],\n        [9.2054057e-01, 3.2171592e-08, 6.7642255e-15, ..., 1.0000000e+00,\n         1.0000000e+00, 1.0000000e+00]], dtype=float32)}"
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import numpy as np\n",
    "\n",
    "lr = interface.learned_parameters()\n",
    "lr"
   ],
   "metadata": {
    "collapsed": false,
    "pycharm": {
     "name": "#%%\n"
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "outputs": [],
   "source": [
    "\n",
    "\n"
   ],
   "metadata": {
    "collapsed": false,
    "pycharm": {
     "name": "#%%\n"
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "outputs": [],
   "source": [],
   "metadata": {
    "collapsed": false,
    "pycharm": {
     "name": "#%%\n"
    }
   }
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 2
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython2",
   "version": "2.7.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 0
}