INNER CODE UNIT · Python

main

google/fully-homomorphic-encryption · demos/cc_fraud/cleartext/estimate_ranges.py:153

def main():
  parser = argparse.ArgumentParser(
      description=(
          "Estimate activation ranges of credit card fraud cleartext model."
      )
  )
  parser.add_argument(
      "--model_path",
      type=str,
      default="demos/cc_fraud/data/mlp_fraud_model_sigmoid.pt",
      help="Path to PyTorch model checkpoint file",
  )
  parser.add_argument(
      "--feature_cols_path",
      type=str,
      default="demos/cc_fraud/data/feature_cols.pkl",
      help="Path to feature columns pickle file",
  )

View source record →