Returns the probability that the given n-gram overlaps with the table. A simple implementation which checks how many tokens in the n-gram are als

Get the model's action(s) from an observation. :param observation: the input observation :param state: The last states (can

A factory method to instantiate data pool

Create the layer that represents the distribution: it will be the logits (flattened) of the MultiCategorical distribution. Yo

The quantile-regression loss, as described in the QR-DQN and TQC papers. Partially taken from https://github.com/bayesgroup/tqc_pytorch.

Resets the episode and returns an observation

Compute the cosine similarity of two vectors. :param vec_hyp: array of dictionary for vector corresponding to hypothesis

Returns first three sentences from the generated text

Filter a distribution of logits using top-k and/or nucleus (top-p) filtering Args: logits: logits distribution shape (batch size, vo

Update policy using the currently gathered rollout buffer.

Update policy using the currently gathered rollout buffer.