Package ai.djl.modality.rl.agent
Interface RlAgent
-
- All Known Implementing Classes:
EpsilonGreedy
,QAgent
public interface RlAgent
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description NDList
chooseAction(RlEnv env, boolean training)
Chooses the next action to take within theRlEnv
.void
trainBatch(RlEnv.Step[] batchSteps)
Trains thisRlAgent
on a batch ofRlEnv.Step
s.
-
-
-
Method Detail
-
chooseAction
NDList chooseAction(RlEnv env, boolean training)
Chooses the next action to take within theRlEnv
.- Parameters:
env
- the current environmenttraining
- true if the agent is currently traning- Returns:
- the action to take
-
trainBatch
void trainBatch(RlEnv.Step[] batchSteps)
Trains thisRlAgent
on a batch ofRlEnv.Step
s.- Parameters:
batchSteps
- the steps to train on
-
-