diff --git a/src/rbm/entity.py b/src/rbm/entity.py
index 734f37e..1e78249 100644
--- a/src/rbm/entity.py
+++ b/src/rbm/entity.py
@@ -47,12 +47,8 @@ class EntityParams:
 	@classmethod
 	def from_dict(cls, params: dict):
 		obj = EntityParams()
-		if "doGaussianVisible" in params:
-			obj.do_gaussian_visible = params["doGaussianVisible"]
-		if "doGaussianHidden" in params:
-			obj.do_gaussian_hidden = params["doGaussianHidden"]
-		if "num_gibbs_samples" in params:
-			obj.num_gibbs_samples = params["num_gibbs_samples"]
+		for key, value in params.items():
+			setattr(obj, key, value)
 
 		return obj