fig, axes = plt.subplots(1, 1, figsize=(15,6)) axes.imshow(np.transpose(classifier.coefs_[target_layer]), cmap=plt.get_cmap("gray"), aspect="auto")