Add arguments to script

This commit is contained in:
s487179 2023-06-13 18:06:50 +02:00
parent 1c573e7a01
commit 3d92c79643

View File

@ -115,11 +115,12 @@ def create_dataset():
home_loan_val_final.to_csv('home_loan_val.csv', index=False)
def main() -> None:
# create_dataset()
create_dataset()
parser = argparse.ArgumentParser(description='A test program.')
parser.add_argument("epochs", help="Prints the supplied argument." ,default='10')
parser.add_argument("--epochs", help="Prints the supplied argument.", default='10')
args = parser.parse_args()
epochs = int(args["epochs"])
config = vars(args)
epochs = int(config["epochs"])
train_dataset = load_data("home_loan_train.csv")
val_dataset = load_data("home_loan_val.csv")