better organization for generalization once we add other methods and such

main.py

@@ -22,12 +22,7 @@ if __name__ == "__main__":
         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"],
     )

neural_network/main.py

@@ -4,7 +4,7 @@ from neural_network.forwardprop import fp
 from neural_network.backprop import bp
 
 
-def get_args():
+def get_args() -> dict:
     """
     returns a dictionary containing
     the arguments to be passed to
@@ -34,16 +34,14 @@ def init(X: np.array, y: np.array, hidden_size: int) -> dict:
 def main(
     X: np.array,
     y: np.array,
-    epochs: int,
-    hidden_size: int,
-    learning_rate: float,
-    activation_func: str,
 ) -> None:
-    wb = init(X, y, hidden_size)
-
-    for e in range(epochs):
+    args = get_args()
+    wb = init(X, y, args["hidden_size"])
 
+    for e in range(args["epochs"]):
         fp()
         bp()
 
         # update weights and biases
+
+    # print results