- simplified TrainingParams.from_dict

- fixed test_rbm
2026-01-06 21:48:09 +01:00
parent 226453c7ae
commit e7b287e632
2 changed files with 5 additions and 13 deletions
@@ -32,17 +32,8 @@ class TrainingParams:
 	@classmethod
 	def from_dict(cls, params: dict):
 		obj = TrainingParams()
-		obj.learning_rate = params["learningRate"]
-		obj.momentum = params["momentum"]
-		obj.weight_decay = params["weightDecay"]
-		obj.l2_lambda = params["l2_lambda"]
-		obj.num_epochs = params["numEpochs"]
-		obj.num_gibbs_samples = params["numGibbs"]
-		obj.mini_batch_size = params["miniBatchSize"]
-		obj.do_rao_blackwell = params["doRaoBlackwell"]
-		obj.do_gibbs_sample_visible = params["gibbsDoSampleVisible"]
-		obj.do_gibbs_sample_hidden = params["gibbsDoSampleHidden"]
-		obj.do_batch_sample = params["doSampleBatch"]
+		for key, value in params.items():
+			setattr(obj, key, value)

 		return obj