updated mimic fitness curve

gkhayes · Mar 9, 2019 · f13d4ad · f13d4ad
1 parent 3fbd890
commit f13d4ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlrose/algorithms.py b/mlrose/algorithms.py
@@ -529,7 +529,7 @@ def mimic(problem, pop_size=200, keep_pct=0.2, max_attempts=10,
  attempts += 1
 
  if curve:
- fitness_curve.append(problem.get_pop_fitness())
+ fitness_curve.append(problem.get_fitness())
 
  best_fitness = problem.get_maximize()*problem.get_fitness()
  best_state = problem.get_state().astype(int)