Eșantionarea din bufferul PER
Înainte de a putea folosi clasa Prioritized Experience Buffer pentru a antrena agentul, mai trebuie să implementezi metoda .sample(). Această metodă primește ca argument dimensiunea eșantionului pe care vrei să-l extragi și returnează tranzițiile eșantionate ca tensors, împreună cu indicii lor din bufferul de memorie și ponderea lor de importanță.
În mediul tău a fost preîncărcat un buffer cu capacitatea 10, din care poți face eșantionarea.
Acest exercițiu face parte din cursul
Deep Reinforcement Learning în Python
Instrucțiuni pentru exercițiu
- Calculează probabilitatea de eșantionare asociată fiecărei tranziții.
- Extrage indicii corespunzători tranzițiilor din eșantion;
np.random.choice(a, s, p=p)extrage un eșantion de dimensiunescu înlocuire din array-ula, pe baza array-ului de probabilitățip. - Calculează ponderea de importanță asociată fiecărei tranziții.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
def sample(self, batch_size):
priorities = np.array(self.priorities)
# Calculate the sampling probabilities
probabilities = ____ / np.sum(____)
# Draw the indices for the sample
indices = np.random.choice(____)
# Calculate the importance weights
weights = (1 / (len(self.memory) * ____)) ** ____
weights /= np.max(weights)
states, actions, rewards, next_states, dones = zip(*[self.memory[idx] for idx in indices])
weights = [weights[idx] for idx in indices]
states_tensor = torch.tensor(states, dtype=torch.float32)
rewards_tensor = torch.tensor(rewards, dtype=torch.float32)
next_states_tensor = torch.tensor(next_states, dtype=torch.float32)
dones_tensor = torch.tensor(dones, dtype=torch.float32)
weights_tensor = torch.tensor(weights, dtype=torch.float32)
actions_tensor = torch.tensor(actions, dtype=torch.long).unsqueeze(1)
return (states_tensor, actions_tensor, rewards_tensor, next_states_tensor,
dones_tensor, indices, weights_tensor)
PrioritizedReplayBuffer.sample = sample
print("Sampled transitions:\n", buffer.sample(3))