4d3ecb451fc1ea0035847f71b003b403f0ffce72,dragonn/vis/__init__.py,,plot_ism,#,225

Before Change

    
    highest_scoring_pos=np.argmax(np.abs(ism_mat),axis=1)
    zero_map=np.zeros(ism_mat.shape)
    for i in range(zero_map.shape[0]):
        zero_map[i][highest_scoring_pos[i]]=1
    product=zero_map*ism_mat
    plt.set_cmap("RdBu")
    axes[0]=plot_bases_on_ax(product,axes[0],show_ticks=False)
    axes[0].set_title(title)

After Change

    assert ism_mat.shape[1]==4

    if x.shape!=2: 
        x=x.squeeze()
    seq_len = x.shape[0]
    product=ism_mat*x
    plt.set_cmap("RdBu")
    axes[0]=plot_bases_on_ax(product,axes[0],show_ticks=False)

Instances