Skip to content

Commit df15494

Browse files
committed
bug-fix and set random seed
1 parent a5d9456 commit df15494

File tree

3 files changed

+6
-3
lines changed

3 files changed

+6
-3
lines changed

cusim/culda/pyculda.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@ def init_model(self):
6363
self.num_words, self.num_docs)
6464

6565
# random initialize alpha and beta
66+
np.random.seed(self.opt.seed)
6667
self.alpha = np.random.uniform( \
6768
size=(self.opt.num_topics,)).astype(np.float32)
6869
self.beta = np.random.uniform( \

cusim/ioutils/pyioutils.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def __init__(self, opt=None):
2323
self.opt = aux.get_opt_as_proto(opt or {}, IoUtilsConfigProto)
2424
self.logger = aux.get_logger("ioutils", level=self.opt.py_log_level)
2525

26-
tmp = tempfile.NamedTemporaryFile(mode='w')
26+
tmp = tempfile.NamedTemporaryFile(mode='w', delete=False)
2727
opt_content = json.dumps(aux.proto_to_dict(self.opt), indent=2)
2828
tmp.write(opt_content)
2929
tmp.close()
@@ -49,7 +49,8 @@ def load_stream_vocab(self, filepath, min_count, keys_path):
4949
self.obj.get_word_vocab(min_count, keys_path)
5050

5151
def convert_stream_to_h5(self, filepath, min_count, out_dir,
52-
chunk_indices=10000):
52+
chunk_indices=10000, seed=777):
53+
np.random.seed(seed)
5354
os.makedirs(out_dir, exist_ok=True)
5455
keys_path = pjoin(out_dir, "keys.txt")
5556
token_path = pjoin(out_dir, "token.h5")
@@ -86,7 +87,7 @@ def convert_stream_to_h5(self, filepath, min_count, out_dir,
8687
cols[offset:offset + data_size] = _cols
8788
vali.resize((offset + data_size,))
8889
vali[offset:offset + data_size] = \
89-
np.uniform(size=(data_size,)).astype(np.float32)
90+
np.random.uniform(size=(data_size,)).astype(np.float32)
9091
indptr[processed:processed + read_lines] = _indptr + offset
9192
offset += data_size
9293
processed += read_lines

cusim/proto/config.proto

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,4 +29,5 @@ message CuLDAConfigProto {
2929
optional int32 epochs = 11 [default = 10];
3030
optional int32 num_iters_in_e_step = 12 [default = 5];
3131
optional double vali_p = 13 [default = 0.2];
32+
optional int32 seed = 14 [default = 777];
3233
}

0 commit comments

Comments
 (0)