Skip to content

Commit cc6cc1e

Browse files
author
Maximilian Reißmann
committed
fix tensor bug
1 parent e10be15 commit cc6cc1e

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

docs/src/examples/tensor-regression.md

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Template based regression
1+
# Tensor Regression
22

33
Hint: This topic is currently under development. The next update will include an improvement in performance and dimension handling.
44

@@ -439,6 +439,5 @@ println(" This demonstrates vector cross product capabilities")
439439

440440
Tensor regression in GeneExpressionProgramming.jl opens up new possibilities for discovering complex mathematical relationships involving higher-dimensional data structures. While computationally more demanding than scalar regression, it provides unique capabilities for applications in physics, engineering, and computer science where tensor operations are fundamental.
441441

442-
443442
---
444443

src/RegressionWrapper.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,7 @@ const GENE_COMMON_PROBS = Dict{String,AbstractFloat}(
268268
"fusion_rate" => 0.0,
269269
"inversion_prob" => 0.1,
270270
"insertion_prob" => 0.1,
271-
"reverse_insertion" => 0.1,
271+
"reverse_insertion" => 0.0,
272272
"reverse_insertion_tail" => 0.0,
273273
"gene_transposition" => 0.1,
274274
"gene_averaging_prob" => 0.0,
@@ -865,7 +865,8 @@ function fit!(regressor::GepTensorRegressor, epochs::Int, population_size::Int,
865865
tourni_size=max(Int(ceil(population_size * 0.003)), 3),
866866
file_logger_callback=file_logger_callback,
867867
save_state_callback=save_state_callback,
868-
load_state_callback=load_state_callback
868+
load_state_callback=load_state_callback,
869+
population_sampling_multiplier=1
869870
)
870871

871872
regressor.best_models_ = best

0 commit comments

Comments
 (0)