Add arguments to script
This commit is contained in:
parent
1c573e7a01
commit
3d92c79643
@ -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")
|
||||
|
Loading…
Reference in New Issue
Block a user