readme++

function_learning.py

@@ -13,14 +13,15 @@ NORMALIZE = True
 NUM_LAYERS = 2
 HIDDEN_DIM = 2
 LEARNING_RATE = 1e-3
-NUM_ITERS = int(8e4)
-RANGE = [-5, 5]
+NUM_ITERS = int(7e4)
+RANGE = [5, 10]
 ARITHMETIC_FUNCTIONS = {
     'add': lambda x, y: x + y,
     'sub': lambda x, y: x - y,
     'mul': lambda x, y: x * y,
     'div': lambda x, y: x / y,
     'squared': lambda x, y: torch.pow(x, 2),
+    'root': lambda x, y: torch.sqrt(x),
 }
 
 
@@ -137,4 +138,3 @@ def main():
 
 if __name__ == '__main__':
     main()
-