Использование якобианской матрицы в solve_ivp - PullRequest
Новая
       8

Использование якобианской матрицы в solve_ivp

0 голосов
/ 30 мая 2018

Я пытаюсь использовать встроенную функцию solve_ivp с матрицей Якоби.Я получаю это сообщение об ошибке:

UserWarning: Следующие аргументы не влияют на выбранный решатель: jac..format ("," .join ("{}". формат (x) для x в посторонних)))

Я не уверен, является ли это синтаксической ошибкой или ошибкой в ​​якобианематрица.

Часть моего кода 

def MyIntFun(t,y):
    return DivWatFlux(y,t,wPar, sPar, RPar,dzIN, dzN)

def jacfunc(t,y):
    jac = Richardsmatrix(y, t, wPar, sPar, RPar, dzIN, dzN)
    return jac

mt.tic()
hwODE = spi.solve_ivp(MyIntFun, [tout[0], tout[-1]], hw0, method='RK45', 
    vectorized=True,rtol=1e-4, jac=jacfunc) 
mt.toc()

Это код для якобиана: 

def Richardsmatrix(hw, t, wPar, sPar, RPar, dzIN, dzN): #the jacobian (a,b,c)

    K=Kfun(hw, wPar, m)
    kRes =RPar.kRobBotR

    a = np.zeros(nN)
    b = np.zeros(nN)
    c = np.zeros(nN)


    ii=np.arange(1, nN-1)

    a[ii] = K[ii, 0]/(dzIN[ii, 0]*dzN[ii-1, 0]) #middle nodes
    a[0] = 0 #a1l is zero and not in matrix 
    a[nN-1] = K[nN-1, 0] / (dzIN[nN-1, 0] * dzN[nN-2, 0]) 

    b[ii] = -K[ii, 0] / (dzIN[ii,0] * dzN[ii-1,0])- K[ii+1,0] / (dzIN[ii,0] * dzN[ii, 0]) #middle nodes
    b[0] = -K[1, 0]/(dzIN[0 , 0]*dzN[0, 0])- kRes / dzIN[0 ,0]
    b[nN-1] = -K[nN-1, 0] / (dzIN[nN-1, 0] * dzN[nN-2, 0])

    c[ii] = K[ii+1,0] / (dzIN[ii,0] * dzN[ii, 0]) #middle nodes
    c[0] = K[1,0] / (dzIN[0, 0] * dzN[0, 0]) 
    c[nN-1] = 0

    B = np.diag(a[1:nN], -1)+ np.diag(b, 0) + np.diag(c[0:nN-1], 1)
    sB = sp.sparse.csc_matrix(B)
    return

Моя группа и я довольно новы в программировании, поэтому любая помощьс благодарностью

1 Ответ

0 голосов
/ 17 июля 2018

Есть две отдельные проблемы.

Во-первых, когда method='RK45' передается solve_ivp, решатель (в данном случае Рунге-Кутта 4/5) не может использовать якобиан.Попробуйте передать solver='Radau', solver='BDF' или solver='LSODA', поскольку они используют якобиан, согласно документации (в частности, задан аргумент ключевого слова jac).

Другая проблема, которую я вижу, состоит в том, что у вашей функции Richardsmatrix есть оператор return, который ничего не возвращает.Вместо этого вы можете попробовать return sB (при условии, что sB - это матрица, которую вы хотите, чтобы функция возвращала).

...