s487179/ium_z487179
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add arguments to script

Browse Source
This commit is contained in:
s487179 2023-06-13 18:06:50 +02:00
parent 1c573e7a01
commit 3d92c79643
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ML/model_train.py
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")