commit c40575d579cb4cc46119c52a5d8eb1d381c78796
parent 5cdb8346df24d0a3b9fb559720bacb6f7ba29a3e
Author: Sondre Wold <sondrewold1@gmail.com>
Date: Mon, 18 Aug 2025 14:00:55 +0200
change hyphen to underscore
Diffstat:
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/experiment.py b/experiment.py
@@ -25,7 +25,7 @@ def parse_args() -> argparse.Namespace:
parser.add_argument("-result_path", type=pathlib.Path)
parser.add_argument("-normalization_value", type=float)
parser.add_argument("-experiment_type", type=str, choices=["sick", "ncip", "ncip-non"])
- parser.add_argument("-model", type=str, choices=["BERT_cls", "BERT_mean", "S_BERT", "S_GPT", "E5", "QWEN"])
+ parser.add_argument("-model", type=str, choices=["BERT-cls", "BERT-mean", "S-BERT", "S-GPT", "E5", "QWEN"])
return parser.parse_args()
@@ -419,7 +419,7 @@ def main() -> None:
model = ModelFactory.get(model_type)
result_path = (
args.result_path
- / f"{args.experiment_type}-{model_type}-lambda{int(args.normalization_value * 10)}"
+ / f"{args.experiment_type}_{model_type}_lambda{int(args.normalization_value * 10)}"
) # shitty logging of lambda parameter, need to fix
experiment = Experiment(
sentence_pairs=sentence_pairs,