simplified EntityParams.from_dict
This commit is contained in:
+2
-6
@@ -47,12 +47,8 @@ class EntityParams:
|
|||||||
@classmethod
|
@classmethod
|
||||||
def from_dict(cls, params: dict):
|
def from_dict(cls, params: dict):
|
||||||
obj = EntityParams()
|
obj = EntityParams()
|
||||||
if "doGaussianVisible" in params:
|
for key, value in params.items():
|
||||||
obj.do_gaussian_visible = params["doGaussianVisible"]
|
setattr(obj, key, value)
|
||||||
if "doGaussianHidden" in params:
|
|
||||||
obj.do_gaussian_hidden = params["doGaussianHidden"]
|
|
||||||
if "num_gibbs_samples" in params:
|
|
||||||
obj.num_gibbs_samples = params["num_gibbs_samples"]
|
|
||||||
|
|
||||||
return obj
|
return obj
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user