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) home_loan_val_final.to_csv('home_loan_val.csv', index=False)
def main() -> None: def main() -> None:
# create_dataset() create_dataset()
parser = argparse.ArgumentParser(description='A test program.') 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() args = parser.parse_args()
epochs = int(args["epochs"]) config = vars(args)
epochs = int(config["epochs"])
train_dataset = load_data("home_loan_train.csv") train_dataset = load_data("home_loan_train.csv")
val_dataset = load_data("home_loan_val.csv") val_dataset = load_data("home_loan_val.csv")