From 10630bbfa7400702cae621139b25435bfd48cf20 Mon Sep 17 00:00:00 2001 From: MarionBWeinzierl Date: Thu, 20 Jul 2023 09:48:17 +0100 Subject: [PATCH] deleted obsolete regression code for #58 --- .../regression/functions.py | 81 ------------------- 1 file changed, 81 deletions(-) delete mode 100644 src/gz21_ocean_momentum/regression/functions.py diff --git a/src/gz21_ocean_momentum/regression/functions.py b/src/gz21_ocean_momentum/regression/functions.py deleted file mode 100644 index 36260769..00000000 --- a/src/gz21_ocean_momentum/regression/functions.py +++ /dev/null @@ -1,81 +0,0 @@ -import numpy as np -from data.pangeo_catalog import get_grid - -# grid_cm26 = get_grid() - -# This defines standard functions used for sparse regression - - -def bz(velocity: np.ndarray): - """ - Return the BZ parameterization - """ - # TODO not efficient to do this every call - grid = ( - grid_cm26.interp(dict(xu_ocean=velocity.xu_ocean, yu_ocean=velocity.yu_ocean)) - * 4 - ) - velocity = velocity / 10 - zeta = ( - velocity["vsurf"].diff(dim="xu_ocean") / grid["dxu"] - - velocity["usurf"].diff(dim="yu_ocean") / grid["dyu"] - ) - d = ( - velocity["usurf"].diff(dim="yu_ocean") / grid["dyu"] - + velocity["vsurf"].diff(dim="xu_ocean") / grid["dxu"] - ) - d_tilda = ( - velocity["usurf"].diff(dim="xu_ocean") / grid["dxu"] - - velocity["vsurf"].diff(dim="yu_ocean") / grid["dyu"] - ) - zeta_sq = zeta**2 - s_x = (zeta_sq - zeta * d).diff(dim="xu_ocean") / grid["dxu"] + ( - zeta * d_tilda - ).diff(dim="yu_ocean") / grid["dyu"] - s_y = (zeta * d_tilda).diff(dim="xu_ocean") / grid["dxu"] + ( - zeta_sq + zeta * d - ).diff(dim="yu_ocean") / grid["dyu"] - k_bt = -4.87 * 1e8 - s_x, s_y = s_x * 1e7 * k_bt, s_y * 1e7 * k_bt - return s_x, s_y - - -def zeta(velocity: np.ndarray): - grid = ( - grid_cm26.interp(dict(xu_ocean=velocity.xu_ocean, yu_ocean=velocity.yu_ocean)) - * 4 - ) - velocity = velocity / 10 - zeta = ( - velocity["vsurf"].diff(dim="xu_ocean") / grid["dxu"] - - velocity["usurf"].diff(dim="yu_ocean") / grid["dyu"] - ) - return zeta * 1e7 - - -def d(velocity: np.ndarray): - grid = ( - grid_cm26.interp(dict(xu_ocean=velocity.xu_ocean, yu_ocean=velocity.yu_ocean)) - * 4 - ) - d = ( - velocity["usurf"].diff(dim="yu_ocean") / grid["dyu"] - + velocity["vsurf"].diff(dim="xu_ocean") / grid["dxu"] - ) - return d * 1e7 - - -def d_tilda(velocity: np.ndarray): - grid = ( - grid_cm26.interp(dict(xu_ocean=velocity.xu_ocean, yu_ocean=velocity.yu_ocean)) - * 4 - ) - d_tilda = ( - velocity["usurf"].diff(dim="xu_ocean") / grid["dxu"] - - velocity["vsurf"].diff(dim="yu_ocean") / grid["dyu"] - ) - return d_tilda * 1e7 - - -def laplacian(velocity: np.ndarray): - pass