def parse_args()

in source/sagemaker/src/package/data_privatization/data_privatization.py [0:0]


def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('--data-dir', type=str, default='/opt/ml/processing/input')
    parser.add_argument('--vectors-dir', type=str, default='/opt/ml/processing/vectors')
    parser.add_argument('--output-dir', type=str, default='/opt/ml/processing/output')
    parser.add_argument('--artifact-output-dir', type=str, default='/opt/ml/processing/artifacts')
    parser.add_argument('--sensitive-filename', type=str, default='train_examples.csv')
    parser.add_argument('--vectors-filename', type=str, default='glove.6B.300d.txt.gz')
    parser.add_argument('--privatized-dir', type=str, default='reviews-privatized')
    parser.add_argument('--sensitive-dir', type=str, default='reviews-sensitive')
    parser.add_argument('--epsilon', type=float, default=23)
    parser.add_argument('--embedding-size', type=int, default=300)

    return parser.parse_args()