Я извлекаю функции через elmo.Train и Test - текстовые данные. Я получаю ошибки при выполнении в Google Colab.Я проверил предыдущие вопросы Stackoverflow, но не смог решить.Точные коды с указателями будут полезны.
elmo = hub.Module("https://tfhub.dev/google/elmo/2", trainable=True)
def elmo_vectors(x):
embeddings = elmo(x.tolist(), signature="default", as_dict=True)["elmo"]
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
sess.run(tf.tables_initializer())
# return average of ELMo features
return sess.run(tf.reduce_mean(embeddings,1))
import tensorflow as tf
import tensorflow_hub as hub
list_train = [train[i:i+100] for i in range(0,train.shape[0],100)]
list_test = [test[i:i+100] for i in range(0,test.shape[0],100)]
# Extract ELMo embeddings
elmo_train = [elmo_vectors(x['clean_tweet']) for x in list_train]
elmo_test = [elmo_vectors(x['clean_tweet']) for x in list_test]
Я получаю следующие ошибки: UnknownError: Не удалось получить алгоритм свертки.Вероятно, это связано с тем, что cuDNN не удалось инициализировать, поэтому попробуйте посмотреть, не было ли напечатано предупреждение в журнале выше.[[node module_apply_default_1 / bilm / CNN_2 / Conv2D_6 (определено в /usr/local/lib/python3.6/dist-packages/tensorflow_hub/native_module.py:517)]] [[Среднее узла (определено в: 8)]]