Come serializzare la funzione sympy lambdified?

Question

Il titolo dice tutto. C'è un modo per serializzare una funzione generato da sympy.lambdify ?:

import sympy as sym 
import pickle 
import dill 
a, b = sym.symbols("a, b") 
expr = sym.sin(a) + sym.cos(b) 
lambdified_expr = sym.lambdify((a, b), expr, modules="numpy") 
pickle.dumps(lambdified_expr) # won't work 
dill.dumps(lambdified_expr) # won't work either 

... Il motivo che voglio fare questo è perché il mio codice genera tante funzioni lambdified ma ho trovato che ci vuole troppo tempo ogni volta .

Answer 1

È effettivamente possibile uso dill a Pickle esso. Le versioni più recenti di dill (ad esempio su github) hanno "impostazioni" che consentono varianti di come il pickle è costruito su dump. Sì, le impostazioni predefinite per dill non riescono su questo oggetto, ma non se si utilizza l'impostazione che traccia ricorsivamente riferimenti globali (ad esempio recurse = True). Questa impostazione è simile a quella fornita da cloudpickle per impostazione predefinita.

>>> import sympy as sym 
>>> import pickle 
>>> import dill 
>>> a, b = symbols("a, b") 
>>> a, b = sym.symbols("a, b") 
>>> expr = sym.sin(a) + sym.cos(b) 
>>> lambdified_expr = sym.lambdify((a, b), expr, modules="numpy") 
>>> 
>>> dill.settings 
{'recurse': False, 'byref': False, 'protocol': 2, 'fmode': 0} 
>>> dill.settings['recurse'] = True 
>>> dill.dumps(lambdified_expr) 
'\x80\x02cdill.dill\n_create_function\nq\x00(cdill.dill\n_unmarshal\nq\x01U\x83c\x02\x00\x00\x00\x02\x00\x00\x00\x03\x00\x00\x00C \x00\x00s\x14\x00\x00\x00t\x00\x00|\x00\x00\x83\x01\x00t\x01\x00|\x01\x00\x83\x01\x00\x17S(\x01\x00\x00\x00N(\x02\x00\x00\x00t\x03\x00\x00\x00sint\x03\x00\x00\x00cos(\x02\x00\x00\x00t\x01\x00\x00\x00at\x01\x00\x00\x00b(\x00\x00\x00\x00(\x00\x00\x00\x00s\x08\x00\x00\x00<string>t\x08\x00\x00\x00<lambda>\x01\x00\x00\x00s\x00\x00\x00\x00q\x02\x85q\x03Rq\x04}q\x05(U\x03cosq\x06cnumpy.core.umath\ncos\nq\x07U\x03sinq\x08cnumpy.core.umath\nsin\nq\tuU\x08<lambda>q\nNN}q\x0btq\x0cRq\r.' 

P.S. Sono l'autore dill, quindi lo saprei.

Answer 2

Infatti, pickle, cPickle e anche aneto falliscono in questo esempio con le impostazioni predefinite.

Ma cloudpickle non fallisce!

pip install cloudpickle 

o

https://github.com/cloudpipe/cloudpickle

import sympy as sym 
from cloudpickle import dumps, loads 
a, b = sym.symbols("a, b") 
expr = sym.sin(a) + sym.cos(b) 
lambdified_expr = sym.lambdify((a, b), expr, modules="numpy") 
var=dumps(lambdified_expr) 
a1=lambdified_expr(10,10) 
del lambdified_expr 
lambdified_expr=loads(var) 
a2=lambdified_expr(10,10) 
a1==a2 # True