Spaces:
Running
Running
MilesCranmer
commited on
Commit
•
fb2f513
1
Parent(s):
893fdd2
Do not store custom loss in Julia main
Browse files- pysr/sr.py +2 -2
pysr/sr.py
CHANGED
@@ -1307,7 +1307,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
|
|
1307 |
complexity_of_operators_str += ")"
|
1308 |
complexity_of_operators = Main.eval(complexity_of_operators_str)
|
1309 |
|
1310 |
-
|
1311 |
|
1312 |
mutationWeights = [
|
1313 |
float(self.weight_mutate_constant),
|
@@ -1331,7 +1331,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
|
|
1331 |
complexity_of_constants=self.complexity_of_constants,
|
1332 |
complexity_of_variables=self.complexity_of_variables,
|
1333 |
nested_constraints=nested_constraints,
|
1334 |
-
loss=
|
1335 |
maxsize=int(self.maxsize),
|
1336 |
hofFile=_escape_filename(self.equation_file_),
|
1337 |
npopulations=int(self.populations),
|
|
|
1307 |
complexity_of_operators_str += ")"
|
1308 |
complexity_of_operators = Main.eval(complexity_of_operators_str)
|
1309 |
|
1310 |
+
custom_loss = Main.eval(self.loss)
|
1311 |
|
1312 |
mutationWeights = [
|
1313 |
float(self.weight_mutate_constant),
|
|
|
1331 |
complexity_of_constants=self.complexity_of_constants,
|
1332 |
complexity_of_variables=self.complexity_of_variables,
|
1333 |
nested_constraints=nested_constraints,
|
1334 |
+
loss=custom_loss,
|
1335 |
maxsize=int(self.maxsize),
|
1336 |
hofFile=_escape_filename(self.equation_file_),
|
1337 |
npopulations=int(self.populations),
|