Neurons example with stim types

Neurons example with stim types

Generate some data

from bayes_window import models, fake_spikes_explore, BayesWindow, BayesRegression
from bayes_window.generative_models import generate_fake_spikes,generate_spikes_stim_types
from importlib import reload
import numpy as np
import altair as alt
df = generate_spikes_stim_types(mouse_response_slope=3,
                                n_trials=2,
                                n_neurons=3,
                                n_mice=4,
                                dur=2, )
window = BayesRegression(df=df, y='isi', treatment='stim', condition=['stim_strength','neuron_x_mouse'],
                              group='mouse')
window.fit(model=models.model_hierarchical)
2021-11-30 15:20:09.126883: E external/org_tensorflow/tensorflow/stream_executor/cuda/cuda_driver.cc:271] failed call to cuInit: CUDA_ERROR_COMPAT_NOT_SUPPORTED_ON_DEVICE: forward compatibility was attempted on non supported HW
2021-11-30 15:20:09.127103: E external/org_tensorflow/tensorflow/stream_executor/cuda/cuda_diagnostics.cc:313] kernel version 450.142.0 does not match DSO version 450.156.0 -- cannot find working devices in this configuration
<bayes_window.slopes.BayesRegression at 0x7f4deec873d0>
window.plot(x='stim_strength',column='mouse',independent_axes=False,row='neuron_x_mouse')
# window.chart_posterior_hdi
window.chart