Old engine for Continuous Time Bayesian Networks. Superseded by reCTBN. 🐍
https://github.com/madlabunimib/PyCTBN
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
598 B
23 lines
598 B
4 years ago
|
import numpy as np
|
||
|
|
|
||
|
|
class NumPyBackend:
|
||
|
|
"""Backend that uses numpy.fft"""
|
||
|
|
__ua_domain__ = "numpy.scipy.fft"
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def __ua_function__(method, args, kwargs):
|
||
|
|
kwargs.pop("overwrite_x", None)
|
||
|
|
|
||
|
|
fn = getattr(np.fft, method.__name__, None)
|
||
|
|
return (NotImplemented if fn is None
|
||
|
|
else fn(*args, **kwargs))
|
||
|
|
|
||
|
|
|
||
|
|
class EchoBackend:
|
||
|
|
"""Backend that just prints the __ua_function__ arguments"""
|
||
|
|
__ua_domain__ = "numpy.scipy.fft"
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def __ua_function__(method, args, kwargs):
|
||
|
|
print(method, args, kwargs, sep='\n')
|