diff options
| -rw-r--r-- | config/lstm-frigo-irc.py | 10 | ||||
| -rwxr-xr-x | ircbot.py | 2 |
2 files changed, 6 insertions, 6 deletions
diff --git a/config/lstm-frigo-irc.py b/config/lstm-frigo-irc.py index 2d0bf3a..4ae21e2 100644 --- a/config/lstm-frigo-irc.py +++ b/config/lstm-frigo-irc.py @@ -3,14 +3,14 @@ from blocks.bricks import Tanh from model.lstm import Model -dataset = 'data/logcompil-2016-03-07.txt' +dataset = 'data/logcompil.txt' io_dim = 256 # An epoch will be composed of 'num_seqs' sequences of len 'seq_len' # divided in chunks of lengh 'seq_div_size' -num_seqs = 100 -seq_len = 2000 -seq_div_size = 100 +num_seqs = 50 +seq_len = 5000 +seq_div_size = 200 hidden_dims = [1024, 1024, 1024] activation_function = Tanh() @@ -26,7 +26,7 @@ l1_reg = 0 step_rule = AdaDelta() # parameter saving freq (number of batches) -monitor_freq = 10 +monitor_freq = 100 save_freq = 100 # used for sample generation and IRC mode @@ -107,7 +107,7 @@ class IRCClient(SimpleIRCClient): else: # feed phrase to bot prob, = pred_f(self.str2data(s0+'\n')) - if self.nick in msg: + if any(x in msg.lower() for x in [self.nick, 'frigal']): self.pred_until(pred_f, prob, '\t') prob, = pred_f(self.str2data(nick+': ')) rep = nick + ': ' + self.pred_until(pred_f, prob) |