Mchristos / empowerment

intrinsic motivation in grid worlds
MIT License
25 stars 1 forks source link

Emp maximising agent always uses visit count and not blahut arimoto #6

Open tessavdheiden opened 4 years ago

tessavdheiden commented 4 years ago

Hi Chris,

please check out this line: https://github.com/Mchristos/empowerment/blob/70a031920e5526263b134df9dbc3070acabbb027/empowerment.py#L146

Shouldn't it be something like: return empowerment(self.T, self.det==1, self.n_step, state, n_samples = self.n_samples)