Merge pull request #23 from thodoris/randint_upperbound

Update base.py
Esse commit está contido em:
robintibor
2018-08-20 13:00:48 +02:00
commit de GitHub
+2 -2
Ver Arquivo
@@ -148,11 +148,11 @@ class BaseModel(object):
self.iterator = CropsFromTrialsIterator(
batch_size=batch_size, input_time_length=input_time_length,
n_preds_per_input=n_preds_per_input,
seed=self.seed_rng.randint(0, 4294967295))
seed=self.seed_rng.randint(0, np.iinfo(np.int32).max-1))
else:
self.iterator = BalancedBatchSizeIterator(
batch_size=batch_size,
seed=self.seed_rng.randint(0, 4294967295))
seed=self.seed_rng.randint(0, np.iinfo(np.int32).max-1))
stop_criterion = MaxEpochs(epochs - 1)# -1 since we dont print 0 epoch, which matters for this stop criterion
train_set = SignalAndTarget(train_X, train_y)
optimizer = self.optimizer