更新src

settings.py

@@ -1,9 +1,9 @@
 import tensorflow as tf
 import numpy as np
+
 flags = tf.app.flags
 FLAGS = flags.FLAGS
 
-
 flags.DEFINE_integer('hidden3', 64, 'Number of units in hidden layer 3.')
 flags.DEFINE_integer('discriminator_out', 0, 'discriminator_out.')
 flags.DEFINE_float('discriminator_learning_rate', 0.001, 'Initial learning rate.')
@@ -21,7 +21,6 @@ infor: number of clusters
 '''
 infor = {'cora': 7, 'citeseer': 6, 'pubmed': 3}
 
-
 '''
 We did not set any seed when we conducted the experiments described in the paper;
 We set a seed here to steadily reveal better performance of ARGA
@@ -30,6 +29,7 @@ seed = 7
 np.random.seed(seed)
 tf.set_random_seed(seed)
 
+
 def get_settings(dataname, model, task):
     if dataname != 'citeseer' and dataname != 'cora' and dataname != 'pubmed':
         print('error: wrong data set name')
@@ -46,4 +46,3 @@ def get_settings(dataname, model, task):
         re = {'data_name': dataname, 'iterations': iterations, 'model': model}
 
     return re
-