From c48bfd2e8c9766416d3070564991cf11fb1596f9 Mon Sep 17 00:00:00 2001 From: TerenceLiu Date: Sun, 19 Feb 2023 12:50:29 +0800 Subject: [PATCH] Update 'algorithm/optimization/gradient.py' --- algorithm/optimization/gradient.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/algorithm/optimization/gradient.py b/algorithm/optimization/gradient.py index c70ece3..5f85bc5 100644 --- a/algorithm/optimization/gradient.py +++ b/algorithm/optimization/gradient.py @@ -364,7 +364,10 @@ class gd2d_compete(object): with self.plot_output: clear_output(wait=True) x1, x2 =symbols("x1 x2") - xx1, xx2 = np.arange(0, 5, 0.25), np.arange(0, 5, 0.25) + if self.wg_expr.value == "(1 - 8 * x1 + 7 * x1**2 - (7/3) * x1**3 + (1/4) * x1**4) * x2**2 * E**(-x2)": + xx1, xx2 = np.arange(0, 5, 0.25), np.arange(0, 5, 0.25) + else: + xx1, xx2 = np.arange(-5, 5, 0.25), np.arange(-5, 5, 0.25) xx1, xx2 = np.meshgrid(xx1, xx2) func = lambdify((x1, x2), sympify(self.wg_expr.value), "numpy") fx = func(xx1, xx2)