import tensorflow as tf

            'input_node': input_node,
            'hidden_layers_node': hidden_layers_node,
            'output_node': output_node,
            'learning_rate': learning_rate,
            'learning_rate_decay': learning_rate_decay,
            'activation': activation,
            'L1_reg': L1_reg,
            'L2_reg': L2_reg,
            'optimizer': optimizer,
            'dropout': dropout_keep_prob
        }
        # Set random state
        tf.set_random_seed(seed)
        # create new Session for the DeepSurv Class
        self.sess = tf.Session(graph=G)
        # Initialize all global variables
        self.sess.run(init_op)

    def train(self, num_epoch=5000, iteration=-1, 
              plot_train_loss=False, plot_train_CI=False):
        """
        Training DeepSurv network.

        Parameters:
