使用 RNN 求解基本數學方程 [附編碼示例]

將張量流導入為tf

從tensorflow.keras.models導入順序

從tensorflow.keras.layers導入Dense、Dropout、SimpleRNN、RepeatVector、TimeDistributed

從tensorflow.keras.callbacks導入EarlyStopping，LambdaCallback

從termcolor導入有色

num_chars = len (char_string)

character_to_index = dict ((c, i) for i, c in enumerate (char_string))

index_to_character = dict ((i, c) for i, c in enumerate (char_string))

如果d != 0則返回n / d否則0

定義數據生成（）：

random1 = np.random.randint（低= 0 ，高= 100 ）

random2 = np.random.randint(low = 0 ,high = 100 )

op = np.random.randint（低= 0 ，高= 4 ）

如果op == 1 ：

arith = str (random1) + ' + ' + str (random2)

res = str (random1 + random2)

elif op == 1 ：

arith = str (random1) + ' – ' + str (random2)

res = str (random1 – random2)

elif op == 2 ：

arith = str (random1) + ' * ' + str (random2)

res = str (random1 * random2)

否則：

arith = str (random1) + ' / ' + str (random2)

res = str ( round (division(random1, random2), 2 ))

返回arith, res

max_time_steps = 5 #我們將輸出硬編碼為 5 個字符

定義模型（）：

模型=順序（）

model.add(SimpleRNN(hidden_​​units, input_shape = ( None , num_chars)))

model.add(RepeatVector(max_time_steps))

model.add(SimpleRNN(hidden_​​units, return_sequences = True ))

model.add(TimeDistributed(Dense(num_chars,activation = ' softmax ' )))

返回模型

模型=模型（）

模型.summary()

model.compile（損失= '分類交叉熵' ，優化器= '亞當' ，指標= [ '準確性' ]）

x = np.zeros((max_time_steps, num_chars))

y = np.zeros((max_time_steps, num_chars))

x_remaining = max_time_steps – len (arith)

y_remaining = max_time_steps – len (res)

對於i, c in enumerate (arith)：

x[x_remaining + i, character_to_index[c]] = 1

對於範圍內的i (x_remaining)：

x[i, character_to_index[ ' 0 ' ]] = 1

對於i, c in enumerate (res)：

y[y_remaining + i, character_to_index[c]] = 1

對於我在範圍內（y_remaining）：

y[i, character_to_index[ ' 0 ' ]] = 1

返回x, y

res = [index_to_character[np.argmax(vec)] for i, vec in enumerate ( input )]

返回'​​ ' .join(res)

標誌=假

輸出= ' '

對於c輸入： _

如果不是標誌和c == ' 0 ' ：

繼續

如果c == ' + '或c == ' – '或c == ' * '或c == ' / '或c == ' 。 ' :

標誌=假

否則：

標誌=真

輸出+= c

返回輸出

x_train = np.zeros((num_equations, max_time_steps, num_chars))

y_train = np.zeros((num_equations, max_time_steps, num_chars))

對於範圍內的i (num_equations)：

e, l =數據生成（）

x, y =矢量化(e, l)

x_train[i] = x

y_train[i] = y

返回x_train, y_train

simple_logger = LambdaCallback(

on_epoch_end = lambda e, l: print ( ' {:.2f} ' .format(l[ ' val_accuracy ' ]), end = ' _ ' )

)

early_stopping = EarlyStopping(monitor = ' val_loss ' , 耐心= 8 )

model.fit（x_train，y_train，epochs = 100 ，validation_split = 0.25 ，verbose = 0 ，

回調= [simple_logger, early_stopping])

preds = model.predict(x_test)

full_seq_acc = 0

對於i, pred in enumerate (preds)：

pred_str =剝離（去矢量化（pred））

y_test_str =剝離（去向量化（y_test[i]））

x_test_str =剝離（去向量化（x_test[i]））

col = ' green ' if pred_str == y_test_str else ' red '

full_seq_acc += 1 / len (preds) * int (pred_str == y_test_str)

outstring = '輸入：{}，輸出：{}，預測：{} ' .format(x_test_str, y_test_str, pred_str)

打印（彩色（outstring，col））

print ( ' \n全序列精度：{:.3f} % ' .format( 100 * full_seq_acc))