- moved learning_rate to args

This commit is contained in:
2023-06-11 21:08:07 +02:00
parent 16529bcac6
commit 27204bd8c8
+1 -3
View File
@@ -22,7 +22,7 @@
## Author: Jens <jens@orion> ## Author: Jens <jens@orion>
## Created: 2023-06-08 ## Created: 2023-06-08
function water_optimizer (TEST_SUITE, k_ve) function water_optimizer (TEST_SUITE, k_ve, learning_rate)
# Load constants # Load constants
water_constants() water_constants()
@@ -76,13 +76,11 @@ T = create_mol([i_Ca i_Mg i_Na i_Cl i_S], [M_Ca_1_mmol M_Mg_1_mmol M_Na_1_mmol M
if TEST_SUITE == 1, if TEST_SUITE == 1,
# Allowed substances, easy # Allowed substances, easy
Pp = [M_EL_Ca; M_EL_Mg; M_EL_Na; M_EL_Cl; M_EL_S]; Pp = [M_EL_Ca; M_EL_Mg; M_EL_Na; M_EL_Cl; M_EL_S];
learning_rate = 0.01;
end end
if TEST_SUITE == 2, if TEST_SUITE == 2,
# Allowed substances, difficult # Allowed substances, difficult
Pp = [M_CaSO4; M_CaCl2; M_MgSO4; M_NaCl; M_HCl]; Pp = [M_CaSO4; M_CaCl2; M_MgSO4; M_NaCl; M_HCl];
learning_rate = 0.001;
end end
# Init vecfit vars # Init vecfit vars