add costum loss fun

Maximilian Reißmann · Maximilian Reißmann · commit f21c0ff88b7c · 2025-07-17T09:29:35.000+10:00
diff --git a/docs/src/examples/basic-regression.md b/docs/src/examples/basic-regression.md
@@ -119,7 +119,7 @@ if hasfield(typeof(regressor), :fitness_history_) &&
    !isnothing(regressor.fitness_history_) && 
    hasfield(typeof(regressor.fitness_history_), :train_loss)
     
-    fitness_history = regressor.fitness_history_.train_loss
+    fitness_history = [elem[1] for elem in regressor.fitness_history_.train_loss]
     
     p1 = plot(1:length(fitness_history), fitness_history,
               xlabel="Generation",
diff --git a/src/RegressionWrapper.jl b/src/RegressionWrapper.jl
@@ -738,7 +738,7 @@ function fit!(regressor::GepRegressor, epochs::Int, population_size::Int, x_trai
         y_train,
         !isnothing(x_test) ? x_test : x_train,
         !isnothing(y_test) ? y_test : y_train,
-        get_loss_function(loss_fun);
+        loss_fun isa String ? get_loss_function(loss_fun) : loss_fun;
         secOptimizer=optimizer_wrapper,
         break_condition=break_condition
     )

Original file line number	Diff line number	Diff line change
`@@ -738,7 +738,7 @@ function fit!(regressor::GepRegressor, epochs::Int, population_size::Int, x_trai`
`738`	`738`	`y_train,`
`739`	`739`	`!isnothing(x_test) ? x_test : x_train,`
`740`	`740`	`!isnothing(y_test) ? y_test : y_train,`
`741`		`- get_loss_function(loss_fun);`
	`741`	`+ loss_fun isa String ? get_loss_function(loss_fun) : loss_fun;`
`742`	`742`	`secOptimizer=optimizer_wrapper,`
`743`	`743`	`break_condition=break_condition`
`744`	`744`	`)`