RewardMonitorAverage.java

Go to the documentation of this file.

package rlpark.plugin.rltoys.experiments.parametersweep.onpolicy.internal;

import rlpark.plugin.rltoys.experiments.helpers.Runner;
import rlpark.plugin.rltoys.experiments.helpers.Runner.RunnerEvent;
import rlpark.plugin.rltoys.experiments.parametersweep.internal.AbstractPerformanceMonitor;
import zephyr.plugin.core.api.signals.Listener;

public class RewardMonitorAverage extends AbstractPerformanceMonitor implements OnPolicyRewardMonitor {
  public RewardMonitorAverage(String prefix, int nbBins, int nbTimeSteps) {
    super(prefix, "Reward", createStartingPoints(nbBins, nbTimeSteps));
  }

  @Override
  public void connect(Runner runner) {
    runner.onTimeStep.connect(new Listener<Runner.RunnerEvent>() {
      @SuppressWarnings("synthetic-access")
      @Override
      public void listen(RunnerEvent runnerEvent) {
        if (runnerEvent.step.time == 0)
          return;
        registerMeasurement(runnerEvent.step.time - 1, runnerEvent.step.r_tp1);
      }
    });
  }

  @Override
  protected double worstValue() {
    return -Float.MAX_VALUE;
  }
}