Найти градиент функции: Sympy vs. Jax - PullRequest
Новая
       59

Найти градиент функции: Sympy vs. Jax

1 голос
/ 20 апреля 2020

У меня есть функция Black_Cox(), которая вызывает другие функции, как показано ниже: 

import numpy as np
from scipy import stats

# Parameters
D = 100
r = 0.05
γ = 0.1

# Normal CDF
N = lambda x: stats.norm.cdf(x)

H = lambda V, T, L, σ: np.exp(-r*T) * N( (np.log(V/L) + (r-0.5*σ**2)*T) / (σ*np.sqrt(T)) )

# Black-Scholes
def C_BS(V, K, T, σ):
    d1 = (np.log(V/K) + (r + 0.5*σ**2)*T ) / ( σ*np.sqrt(T) )
    d2 = d1 - σ*np.sqrt(T)
    return V*N(d1) - np.exp(-r*T)*K*N(d2)

def BL(V, T, D, L, σ):
    return L * H(V, T, L, σ) - L * (L/V)**(2*r/σ**2-1) * H(L**2/V, T, L, σ) \
              + C_BS(V, L, T, σ) - (L/V)**(2*r/σ**2-1) * C_BS(L**2/V, L, T, σ) \
              - C_BS(V, D, T, σ) + (L/V)**(2*r/σ**2-1) * C_BS(L**2/V, D, T, σ)

def Bb(V, T, C, γ, σ, a):
    b = (np.log(C/V) - γ*T) / σ
    μ = (r - a - 0.5*σ**2 - γ) / σ
    m = np.sqrt(μ**2 + 2*r)
    return C*np.exp(b*(μ-m)) * ( N((b-m*T)/np.sqrt(T)) + np.exp(2*m*b)*N((b+m*T)/np.sqrt(T)) )

def Black_Cox(V, T, C=160, σ=0.1, a=0):
    return np.exp(γ*T)*BL(V*np.exp(-γ*T), T, D*np.exp(-γ*T), C*np.exp(-γ*T), σ) + Bb(V, T, C, γ, σ, a)

Мне нужно работать с производной функции Black_Cox относительно V. Точнее, мне нужно оценить эту производную по тысячам путей, где я меняю другие аргументы, нахожу производную и оцениваю по какому-то V.

Каков наилучший способ продолжить?

  • Должен ли я использовать sympy, чтобы найти эту производную, а затем оценить по своему выбору V, как я бы сделал в Mathematica: D[BlackCox[V, 10, 100, 160], V] /. V -> 180 или

  • Должен ли я просто использовать jax?

Если sympy, как бы вы посоветовали мне сделать это?

С jax Я понимаю, что мне нужно выполнить следующие операции импорта: 

import jax.numpy as np
from jax.scipy import stats
from jax import grad

и переоценить мои функции до получения градиента: 

func = lambda x: Black_Cox(x,10,160,0.1)
grad(func)(180.0)

Если мне все еще потребуется работать с numpy версией функции, мне нужно будет создать 2 экземпляра каждой функции или есть элегантный способ дублировать функцию для jax целей?

1 Ответ

1 голос
/ 04 мая 2020

Jax не предоставляет никакого встроенного способа перекомпиляции функции numpy с использованием версий jax numpy и scipy. Но вы можете использовать фрагмент, подобный следующему, чтобы сделать это автоматически: 

import inspect
from functools import wraps
import numpy as np
import jax.numpy

def replace_globals(func, globals_):
  """Recompile a function with replaced global values."""
  namespace = func.__globals__.copy()
  namespace.update(globals_)
  source = inspect.getsource(func)
  exec(source, namespace)
  return wraps(func)(namespace[func.__name__])

Он работает так: 

def numpy_func(N):
  return np.arange(N) ** 2

jax_func = replace_globals(numpy_func, {"np": jax.numpy})

Теперь вы можете оценить версию numpy: 

numpy_func(10)
# array([ 0,  1,  4,  9, 16, 25, 36, 49, 64, 81])

и версия jax: 

jax_func(10)
# DeviceArray([ 0,  1,  4,  9, 16, 25, 36, 49, 64, 81], dtype=int32)

Просто убедитесь, что вы заменяете все соответствующие глобальные переменные при переносе более сложной функции.

...