I wanna pack relevant functions into class, as bound member(belongto instance and can accept “self” parameter)
Yes, I always make my code in the form of:
class ProbModule:
def __init__(self, f1, f2):
# assignments etc.
self.f1 = f1
self.f2 = f2
def model(self, *args):
...
def guide(self, *args):
...
def infer(self, *args):
...