mirror of https://github.com/kubeflow/examples.git
Update housing.py
This commit is contained in:
parent
a85cc31fce
commit
c38a19c8ea
|
@ -81,7 +81,7 @@ def main(args):
|
|||
eval_model(model, test_X, test_y)
|
||||
save_model(model, args.model_file)
|
||||
|
||||
if __name__=='__main__':
|
||||
if __name__ == '__main__':
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument(
|
||||
'--train-input',
|
||||
|
|
Loading…
Reference in New Issue