Skip to content

Commit 1cd053e

Browse files
committed
Fixed GP fitness again
1 parent 7bf3f4c commit 1cd053e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

causal_testing/estimation/genetic_programming_regression_fitter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -294,7 +294,7 @@ def fitness(self, expression: gp.PrimitiveTree) -> float:
294294
sqerrors = (self.df[self.outcome] - y_estimates) ** 2
295295
nrmse = np.sqrt(sqerrors.sum() / len(self.df)) / (self.df[self.outcome].max() - self.df[self.outcome].min())
296296

297-
if pd.isnull(nrmse) or nrmse.real != nrmse:
297+
if pd.isnull(nrmse) or nrmse.real != nrmse or y_estimates.dtype != self.df.dtypes[self.outcome]:
298298
return (float("inf"),)
299299

300300
return (nrmse,)

0 commit comments

Comments
 (0)