TypeError при нахождении градиента скаляра в автограде

Возможно, это очевидно, но я не могу заставить автограда работать над поиском градиентов для скаляра. Приведенный ниже код выдает "*** TypeError: ufunc 'xlogy', не поддерживаемый для типов ввода, ..." Что не так с типом переменной p?

import autograd.numpy as np
from autograd.scipy.special import scipy
from autograd import grad
from scipy.optimize import minimize

def logpmf(p):
    k = 33
    n = 100
    return -scipy.stats.binom.logpmf(k, n, p)

x0 = np.array([0.0])
result = minimize(logpmf, x0=x0, jac=grad(logpmf), method='CG')
print("Found minimum at {0}".format(result.x))

0 ответов

Другие вопросы по тегам