Я хочу попробовать пример LSTM онлайн, но я сталкиваюсь с некоторыми проблемами. Я впервые использую YAML.
Я запускаю код, включая приведенные ниже части, и появляются следующие ошибки:
Traceback (most recent call last):
File "lstm_test.py", line 112, in <module>
lstm_predict(strings)
File "lstm_test.py", line 74, in lstm_predict
yaml.warnings({'YAMLLoadWarning': False})
AttributeError: module 'yaml' has no attribute 'warnings'
Я запускаю код в Windows 10 и использую Python 3.7.
def lstm_predict(strings):
print('loading model......')
with open('../model/lstm.yml', 'r') as f:
yaml_string = yaml.load(f)
model = model_from_yaml(yaml_string)
print('loading weights......')
model.load_weights('../model/lstm.h5')
model.compile(loss='categorical_crossentropy', optimizer='adam',metrics=['accuracy'])
for string in strings:
line = Converter('zh-hant').convert(string.encode().decode('utf-8'))
string0 = line.encode('utf-8')
print("="*20)
data=input_transform(string0)
data.reshape(1,-1)
#print data
result=model.predict_classes(data)
print(result) # [[1]]
if result[0]==1:
print(string,' positive')
elif result[0]==0:
print(string,' neural')
else:
print(string,' negative')
На самом деле я получаю ошибку:
AttributeError: module 'yaml' has no attribute 'warnings'.
Но теоретически я думаю, что это не должно вызывать каких-либо ошибок. Что я могу пропустить?