From 647dce75ba7483b8fa0b64ef64ecaab70932b769 Mon Sep 17 00:00:00 2001 From: Alex Auvolat Date: Mon, 25 Apr 2016 10:58:34 +0200 Subject: Batch size changes, mostly --- config/lstm-frigo-irc.py | 10 +++++----- 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 diff --git a/ircbot.py b/ircbot.py index ac329eb..7477691 100755 --- a/ircbot.py +++ b/ircbot.py @@ -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) -- cgit v1.2.3