Pythonでcannot import nameエラーが出ます。

Question

see-rnnというライブラリを利用しようとしているのですが、GitのサンプルのPythonコードは動作するのですが、このライブラリを自作のスクリプトに埋め込んだところ、

from see_rnn import get_rnn_gradients, show_features_1D, show_features_2D
ImportError: cannot import name 'get_rnn_gradients' from 'see_rnn' (C:\Users\username\AppData\Local\Programs\Python\Python37\lib\site-packages\see_rnn_init_.py)

というエラーメッセージが出て動作しません。

#seernn
from keras.layers import Input, LSTM
from keras.models import Model
from keras.optimizers import Adam
from see_rnn import get_rnn_gradients, show_features_1D, show_features_2D
from see_rnn import show_features_0D

このfrom see_rnnの行でエラーが発生します。

よろしくお願いいたします。
サンプルは

import numpy as np
from keras.layers import Input, LSTM
from keras.models import Model
from keras.optimizers import Adam
from see_rnn import get_rnn_gradients, show_features_1D, show_features_2D
from see_rnn import show_features_0D

def make_model(rnn_layer, batch_shape, units):
    ipt = Input(batch_shape=batch_shape)
    x   = rnn_layer(units, activation='tanh', return_sequences=True)(ipt)
    out = rnn_layer(units, activation='tanh', return_sequences=False)(x)
    model = Model(ipt, out)
    model.compile(Adam(4e-3), 'mse')
    return model
    
def make_data(batch_shape):
    return np.random.randn(*batch_shape), \
           np.random.uniform(-1, 1, (batch_shape[0], units))

def train_model(model, iterations, batch_shape):
    x, y = make_data(batch_shape)
    for i in range(iterations):
        model.train_on_batch(x, y)
        print(end='.')  # progbar
        if i % 40 == 0:
            x, y = make_data(batch_shape)

units = 6
batch_shape = (16, 100, 2*units)

model = make_model(LSTM, batch_shape, units)
train_model(model, 300, batch_shape)

x, y  = make_data(batch_shape)
grads_all  = get_rnn_gradients(model, x, y, layer_idx=1)  # return_sequences=True
grads_last = get_rnn_gradients(model, x, y, layer_idx=2)  # return_sequences=False

show_features_1D(grads_all, n_rows=2, show_xy_ticks=[1,1])
show_features_2D(grads_all, n_rows=8, show_xy_ticks=[1,1], norm=(-.01, .01))
show_features_0D(grads_last)

です。

追記：
https://github.com/OverLordGoldDragon/see-rnn/issues/9
にある通り、不具合のようです。

Comments

[meg_]

動いた「GitのサンプルのPythonコード」とimportしているものは同じなのでしょうか？

[yutapatio]

コメントありがとうございます。
問題が起きているコードの方では下記のimportをしてます。
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras.utils import to_categorical
from tensorflow.keras import models,layers, regularizers
from tensorflow.keras.layers import Dense, Activation, BatchNormalization, Dropout, Flatten, LSTM, Embedding
from tensorflow.keras.models import Sequential
from keras.utils import plot_model
from sklearn.model_selection import train_test_split
import matplotlib.pylab as plt
from sklearn import datasets, model_selection, metrics
#seernn
from keras.layers import Input, LSTM
from keras.models import Model
from keras.optimizers import Adam
from see_rnn import get_rnn_gradients, show_features_1D, show_features_2D
from see_rnn import show_features_0D

[meg_]

下記と同じ症状ではないでしょうか？　不具合かもしれませんね。
https://github.com/OverLordGoldDragon/see-rnn/issues/9

[yutapatio]

ありがとうございます。上記URL私のポスティングに対する作者の回答でして、確かに不具合のようです。どうもありがとうございました。