From e244104e7b697b79e8500b5d648ec161c4ed9a63 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sat, 8 Jul 2023 13:59:28 +0200 Subject: [PATCH] Update. --- expr.py | 3 ++- main.py | 2 +- tasks.py | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/expr.py b/expr.py index e539fcb..7b31b8c 100755 --- a/expr.py +++ b/expr.py @@ -44,6 +44,7 @@ def random_expr(variables, budget): def generate_program(nb_variables, length): s = "" variables = set() + length = min(length, 1+torch.randint(length*2, (1,)).item()) while len(s) < length: v = random_var(nb_variables=nb_variables) s += v + "=" + random_expr(variables, budget=20) + ";" @@ -86,7 +87,7 @@ if __name__ == "__main__": import time start_time = time.perf_counter() - sequences = generate_sequences(1000, length=30) + sequences = generate_sequences(1000, length=40) end_time = time.perf_counter() for s in sequences[:10]: print(s) diff --git a/main.py b/main.py index 56b7e1c..e2b705d 100755 --- a/main.py +++ b/main.py @@ -125,7 +125,7 @@ parser.add_argument("--stack_fraction_values_for_train", type=float, default=0.7 parser.add_argument("--expr_nb_variables", type=int, default=5) -parser.add_argument("--expr_sequence_length", type=int, default=30) +parser.add_argument("--expr_sequence_length", type=int, default=40) parser.add_argument("--expr_input_file", type=str, default=None) diff --git a/tasks.py b/tasks.py index 463d94c..cec6704 100755 --- a/tasks.py +++ b/tasks.py @@ -911,7 +911,7 @@ class Expr(Task): test_nb_correct, test_nb_delta, test_nb_missed, - ) = compute_nb_correct(self.test_input[:1000]) + ) = compute_nb_correct(self.test_input[:10000]) logger( f"accuracy_test {n_epoch} nb_total {test_nb_total} nb_correct {test_nb_correct} accuracy {(100.0*test_nb_correct)/test_nb_total:.02f}%" -- 2.39.5