From fb4df2ddc83cb014636a38a39647f49ef8acd2ab Mon Sep 17 00:00:00 2001 From: Ashish Patel Date: Thu, 21 Oct 2021 15:51:24 +0530 Subject: [PATCH] udpated ```python tf.train.AdamOptimizer =>tf.keras.optimizers.Adam() ``` --- GNN.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/GNN.py b/GNN.py index 8ef4dd6..0659fd4 100644 --- a/GNN.py +++ b/GNN.py @@ -4,7 +4,7 @@ # class for the core of the architecture class GNN: - def __init__(self, net, input_dim, output_dim, state_dim, max_it=50, optimizer=tf.train.AdamOptimizer, learning_rate=0.01, threshold=0.01, graph_based=False, + def __init__(self, net, input_dim, output_dim, state_dim, max_it=50, optimizer=tf.keras.optimizers.Adam(), learning_rate=0.01, threshold=0.01, graph_based=False, param=str(time.datetime.now().strftime('%Y-%m-%d_%H-%M-%S')), config=None, tensorboard=False, mask_flag=False): """ create GNN instance. Feed this parameters: