extend model member detection by list[Entity]
This commit is contained in:
@@ -1,6 +1,8 @@
|
||||
import os
|
||||
from abc import ABC, abstractmethod
|
||||
|
||||
from IPython.utils.wildcard import is_type
|
||||
|
||||
from rbm.entity import Entity
|
||||
from rbm.train import train
|
||||
from rbm.matrix import Mat, np
|
||||
@@ -18,6 +20,8 @@ class Model(ABC):
|
||||
for name, value in self.__dict__.items():
|
||||
if isinstance(value, obj_type):
|
||||
obj_list.append(value)
|
||||
elif isinstance(value, type(obj_list)):
|
||||
obj_list += value
|
||||
return obj_list
|
||||
|
||||
def train(self, batch: Mat, status: Status = None):
|
||||
|
||||
Reference in New Issue
Block a user