def multiVariable1():
'''
Tests Gradient Descent with multiple variables
'''
x = AD.AADVariable(3, [1, 0])
y = AD.AADVariable(2, [0, 1])
solve = AAD_grad.solve(lambda X: AD.abs(AD.cos(X[0]) * X[1] - 12), [x, y],
0.001,
progress_step=None,
max_iter=10000)
assert (solve[0].val < tol and solve[1].val - 12 < tol)
def singleVariable():
'''
Tests Gradient Descent with a single variable and a single vector
'''
x = AD.AADVariable(3, [1, 0])
y = AD.AADVariable(2, [0, 1])
solve = AAD_grad.solve(lambda X: AD.abs(2 * X[0] - 12), [x, x],
0.001,
progress_step=None,
max_iter=10000)
assert (solve[0].val - 6) < tol
def multiVariable2():
'''
Tests Gradient Descent with multiple variables
'''
x = AD.AADVariable(3, [1, 0])
y = AD.AADVariable(2, [0, 1])
solve = AAD_grad.solve(lambda X: AD.abs(2 * X[0] - 12 + 2 * X[1]**2),
[x, y],
0.001,
progress_step=5000,
max_iter=10000)
assert (solve[0].val - 2.972 < tol and solve[1].val - 1.745 < tol)