DataPredict

API Reference - Others - WassersteinGenerativeAdversarialImitationLearning (WGAIL)

WassersteinGenerativeAdversarialImitationLearning allows an agent to learn from experts’ trajectories.

Notes

• The ReinforcementLearning and Discriminator models must be created separately. Then use setReinforcementLearningModel() and setDiscriminatorModel() to put it inside the GenerativeAdversarialImitationLearning model.

• ReinforcementLearning and Discriminator models must be a part of NeuralNetwork model. If you decide to use linear regression or logistic regression, then it must be constructed using NeuralNetwork model.

• Ensure the output layer of the Discriminator model has only one neuron and its activation function set to “Sigmoid”. It is the default setting for all Discriminator models in research papers.

• The number of neurons at the ReinforcementLearning’s output layer must be equal to the number of neurons at the Discriminator’s input layer.

• It is recommended that the learning rate for the ReinforcementLearning is higher than the Discriminator.

Constructors

new()

Create new model object. If any of the arguments are nil, default argument values for that argument will be used.

WassersteinGenerativeAdversarialImitationLearning.new(numberOfStepsPerEpisode: number): ModelObject

Parameters:

• numberOfStepsPerEpisode: How many steps are needed for it to be considered as a single episode.

Returns:

• ModelObject: The generated model object.

Functions

setParameters()

Set model’s parameters. When any of the arguments are nil, previous argument values for that argument will be used.

WassersteinGenerativeAdversarialImitationLearning:setParameters(numberOfStepsPerEpisode: number)

Parameters:

• numberOfStepsPerEpisode: How many steps are needed for it to be considered as a single episode.

setReinforcementLearningModel()

Sets the ReinforcementLearning into the model.

WassersteinGenerativeAdversarialImitationLearning:setReinforcementLearningModel(ReinforcementLearningModel: Model)

Parameters:

• ReinforcementLearningModel: The model to be used for mimicking the expert.

setDiscriminatorModel()

Sets the Discriminator into the model.

WassersteinGenerativeAdversarialImitationLearning:setDiscriminatorModel(DiscriminatorModel: Model)

Parameters:

• Discriminator: The model to be used for discriminating real and fake contents.

getReinforcementLearningModel()

Gets the ReinforcementLearning from the model.

WassersteinGenerativeAdversarialImitationLearning:getReinforcementLearningModel(): Model

Returns:

• GeneratorModel: The model used for generating contents out of random noise.

getDiscriminatorModel()

Gets the Discriminator from the model.

WassersteinGenerativeAdversarialImitationLearning:getDiscriminatorModel(): Model

Returns:

• DiscriminatorModel: The model used for discriminating real and fake contents.

setClassesList()

OneVsAll:setClassesList(ClassesList: [])

Parameters:

• ClassesList: A list of classes. The index of the list relates to which model belong to. For example, {3, 1} means that the output for 3 is at first model, and the output for 1 is at second model.

getClassesList()

OneVsAll:getClassesList(): []

Returns:

• ClassesList: A list of classes. The index of the list relates to which model belong to. For example, {3, 1} means that the output for 3 is at first model, and the output for 1 is at second model.

categoricalTrain()

Categorically trains the model.

WassersteinGenerativeAdversarialImitationLearning:categoricalTrain(previousFeatureMatrix: matrix, expertActionMatrix: matrix, currentFeatureMatrix: matrix)

Parameters:

• previousFeatureMatrix: The matrix containing the feature environment values.

• expertActionMatrix: The matrix containing the action values generated by an expert.

• currentFeatureMatrix: The matrix containing the feature environment values after an action has been taken by the expert.

diagonalGaussianTrain()

Diagonally Gaussian trains the model.

WassersteinGenerativeAdversarialImitationLearning:diagonalGaussianTrain(previousFeatureMatrix: matrix, expertActionMeanMatrix: matrix, expertStandardDeviationMatrix: matrix,currentFeatureMatrix: matrix)

Parameters:

• previousFeatureMatrix: The matrix containing the feature environment values.

• expertActionMeanMatrix: The matrix containing the action mean values generated by an expert.

• expertStandardDeviationMatrix: The matrix containing the action standard deviation values generated by an expert.

• currentFeatureMatrix: The matrix containing the feature environment values after an action has been taken by the expert.

evaluate()

Generates the output from Discriminator.

WassersteinGenerativeAdversarialImitationLearning:evaluate(featureMatrix: matrix): matrix

Parameters:

• featureMatrix: The matrix containing the feature environment values.

Returns:

• outputMatrix: The matrix containing all the output values.

generate()

Generates the output from Generator.

WassersteinGenerativeAdversarialImitationLearning:generate(featureMatrix: matrix, returnOriginalOutput: boolean): matrix 

Parameters:

• featureMatrix: The matrix containing the feature environment values.

Returns:

• actionVector: The vector containing the actions with the highest values.

-OR-

• actionMatrix: The matrix containing all the action values.

References

• Generative Adversarial Imitation Learning

• Generative Adversarial Networks