Skip to content

Commit 8db4acf

Browse files
committed
Passing training data to algorithm.model in test_population
1 parent 62f6549 commit 8db4acf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

experiment/test_population.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def test_population(ml):
5858

5959
population = algorithm.get_population(algorithm.est)
6060
best_model = algorithm.get_best_solution(algorithm.est)
61-
print(algorithm.model(best_model))
61+
print(algorithm.model(best_model, X_train))
6262
print(algorithm.est.predict(X_train))
6363

6464
# assert that population has at least 1 and no more than 100 individuals

0 commit comments

Comments
 (0)