Numpy-Neuron / main.py
Jensen-holm's picture
just focusing on neural_network first
f9522cf
raw
history blame
760 Bytes
import numpy as np
import neural_network.main as nn
def random_dataset():
"""
initializes a training and
a testing dataset in the form
of numpy arrays
"""
np.random.seed(8675309)
return (
np.random.randn(10000, 10),
np.random.randint(5, size=(10000, 1)),
)
if __name__ == "__main__":
method = input("\nChoose a method to test: ").lower()
if method != "nn":
raise ValueError(f"Invalid method '{method}'. Choose 'nn' instead.")
X, y = random_dataset()
args = nn.get_args()
nn.main(
X=X,
y=y,
epochs=args["epochs"],
hidden_size=args["hidden_size"],
learning_rate=args["learning_rate"],
activation_func=args["activation_func"],
)