meehai

Untitled

Sep 16th, 2020
765
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. from argparse import ArgumentParser
  2.  
  3. def getArgs():
  4.     parser = ArgumentParser()
  5.     parser.add_argument("type")
  6.     parser.add_argument("datasetPath")
  7.     parser.add_argument("--weightsFile")
  8.     args = parser.parse_args()
  9.     return args
  10.  
  11. def main():
  12.     args = getArgs()
  13.  
  14.     if args.type == "train":
  15.         pass
  16.     elif args.type == "test":
  17.         pass
  18.     elif args.type == "export_csv":
  19.         pass
  20.     else
  21.         assert False
  22.  
  23. if __name__ == "__main__":
  24.     main()
RAW Paste Data