From e7fec72e96856711858fc6a911839daf6b4ac51b Mon Sep 17 00:00:00 2001
From: Christian Geishauser <christian.geishauser@hhu.de>
Date: Thu, 28 Oct 2021 09:53:38 +0000
Subject: [PATCH] Update description

---
 policy/FeudalGainPolicy.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/policy/FeudalGainPolicy.py b/policy/FeudalGainPolicy.py
index 5ea4b20..71236ca 100644
--- a/policy/FeudalGainPolicy.py
+++ b/policy/FeudalGainPolicy.py
@@ -21,13 +21,17 @@
 ###############################################################################
 
 
-"""
-Implementation of FeudalGain
-
-Paper: Arxiv reference
+'''
+FeudalGainPolicy.py - What Does The User Want? Information Gain for Hierarchical Dialogue Policy Optimisation
+==================================================
 
 Author: Christian Geishauser
-"""
+
+The implementation of the FeudalGain algorithm that incorporates information gain as intrinsic reward in order to update a Feudal policy.
+The details can be found here: https://arxiv.org/abs/2109.07129
+
+'''
+
 
 import numpy as np
 import random
-- 
GitLab