| 93 | } |
| 94 | |
| 95 | std::vector<double> QLearn::getModelRewards(State state) { |
| 96 | std::vector<double> rewards; |
| 97 | std::for_each(models.begin(), models.end(), [&](Model model){ |
| 98 | rewards.push_back(model.network->getOutput(state)[0]); |
| 99 | }); |
| 100 | |
| 101 | return rewards; |
| 102 | } |
| 103 | |
| 104 | double QLearn::getHighestReward(State state) { |
| 105 | std::vector<double> rewards = getModelRewards(state); |