# Imports
%matplotlib inline
%config InlineBackend.figure_format = 'retina'
from basepair.imports import *
hv.extension('bokeh')
# Common paths
model_dir = Path(f"{ddir}/processed/chipnexus/exp/models/osnk-pstat-sall-smad-zfp/models/best/")
modisco_dir = model_dir / f"modisco/deeplift/profile/"
from basepair.cli.imp_score import ImpScoreFile
imp_score = ImpScoreFile(model_dir / 'deeplift.all.h5')
contrib_scores = imp_score.get_contrib()
from kipoi.data_utils import get_dataset_item
from basepair.plot.tracks import filter_tracks
N = len(imp_score)
plot_tracks(filter_tracks(get_dataset_item(contrib_scores, 10), [400, 600]))
idx_list = pd.Series(np.arange(N)).sample(10)
for idx in idx_list:
plot_tracks(filter_tracks(get_dataset_item(contrib_scores, idx), [400, 600]),
fig_height_per_track=.8, same_ylim=True, rotate_y=0, fig_width=15)