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