X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=expr.py;h=e539fcbb8ec58c17a33dd4686669546a17713df9;hb=233f57347c9560aec2f3cbaf001a8efa56a0243b;hp=818360b16f4ac008b15dd9db049981f7f9f7c677;hpb=5aee50805cfad1dd49bbf30b30fe65b05e03de78;p=picoclvr.git

diff --git a/expr.py b/expr.py
index 818360b..e539fcb 100755
--- a/expr.py
+++ b/expr.py
@@ -24,7 +24,7 @@ def random_expr(variables, budget):
         else:
             return str(torch.randint(10, (1,)).item())
     else:
-        op = torch.randint(4, (1,)).item()
+        op = torch.randint(3, (1,)).item()
         if op == 0:
             e = random_expr(variables, budget - 2)
             if ("+" in e or "-" in e or "*" in e) and (e[0] != "(" or e[-1] != ")"):
@@ -38,8 +38,6 @@ def random_expr(variables, budget):
             if op == 1:
                 return e1 + "+" + e2
             elif op == 2:
-                return e1 + "+" + e2
-            elif op == 3:
                 return e1 + "*" + e2