exploitintel commited on
Commit
dbe2ea6
·
verified ·
1 Parent(s): 4dc95b5

Update default dataset id to exploitintel

Browse files
Files changed (1) hide show
  1. evaluate.py +1 -1
evaluate.py CHANGED
@@ -129,7 +129,7 @@ def score(truths: list[set[str]], preds: list[set[str]], easies: list[bool]) ->
129
  def main() -> None:
130
  ap = argparse.ArgumentParser(description="Evaluate a CVE->CWE model on the test split.")
131
  ap.add_argument("--model", required=True, help="path or HF id of the fine-tuned (merged) model")
132
- ap.add_argument("--dataset", default="eiphuggincve/cve-cwe-consensus")
133
  ap.add_argument("--split", default="test")
134
  ap.add_argument(
135
  "--limit", type=int, default=None, help="evaluate only the first N rows (quick check)"
 
129
  def main() -> None:
130
  ap = argparse.ArgumentParser(description="Evaluate a CVE->CWE model on the test split.")
131
  ap.add_argument("--model", required=True, help="path or HF id of the fine-tuned (merged) model")
132
+ ap.add_argument("--dataset", default="exploitintel/cve-cwe-consensus")
133
  ap.add_argument("--split", default="test")
134
  ap.add_argument(
135
  "--limit", type=int, default=None, help="evaluate only the first N rows (quick check)"