【发布时间】:2020-02-18 07:46:46
【问题描述】:
刚刚用 Anaconda 安装了 Cuda。当尝试运行安装前工作的相同模型时,我在第一次添加 lstm 层时收到错误消息:TypeError: ('Keyword argument not understand:', 'activation')。
我的代码如下所示:
from tensorflow.keras.layers import Dense, Activation, Embedding, LSTM, Dropout, CuDNNLSTM
from tensorflow.keras.models import Sequential
import tensorflow as tf
from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
import pylab as pl
import seaborn as sns
model = Sequential()
model.add(CuDNNLSTM(128, input_shape=(800,1), activation='tanh', return_sequences=True))
model.add(Dropout(0.2))
model.add(CuDNNLSTM(128, activation='tanh'))
model.add(Dropout(0.2))
model.add(Dense(32, activation='tanh'))
model.add(Dropout(0.2))
model.add(Dense(1, activation='sigmoid'))
【问题讨论】:
标签: python tensorflow