- StackFactory removed default constructor
- StackFactory added type hints
This commit is contained in:
@@ -1,19 +1,16 @@
|
|||||||
import json
|
import json
|
||||||
from collections.abc import Callable
|
from collections.abc import Callable
|
||||||
from stack import StackType
|
from stack import Stack, StackType
|
||||||
from layer import Layer
|
from layer import Layer
|
||||||
from params import RbmParams
|
from params import RbmParams
|
||||||
|
|
||||||
from stack_deep import StackDeep
|
from stack_deep import StackDeep
|
||||||
from stack_rnn import StackRnn
|
from stack_rnn import StackRnn
|
||||||
|
|
||||||
|
|
||||||
class StackFactory:
|
class StackFactory:
|
||||||
def __init__(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_dict(cls, project: dict, layer_constructor: Callable = None):
|
def from_dict(cls, project: dict, layer_constructor: Callable = None) -> Stack:
|
||||||
name = project["stack"]["name"]
|
name = project["stack"]["name"]
|
||||||
layers = project["stack"]["layers"]
|
layers = project["stack"]["layers"]
|
||||||
|
|
||||||
@@ -59,7 +56,7 @@ class StackFactory:
|
|||||||
return obj
|
return obj
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_file(cls, filename: str, layer_constructor: Callable = None):
|
def from_file(cls, filename: str, layer_constructor: Callable = None) -> Stack:
|
||||||
obj = None
|
obj = None
|
||||||
with open(filename, "r") as fp:
|
with open(filename, "r") as fp:
|
||||||
prj = json.load(fp)
|
prj = json.load(fp)
|
||||||
|
|||||||
Reference in New Issue
Block a user