simpleNetArch

This function is a helper function for generating the network architecture parameter. A list object used as input for the node_map parameter is returned.

Parameters:

  • x_train - training set matrix. Expected numpy array.
  • n_nodes - number of nodes desired in the hidden layer.