Skip to content

Commit ea2443b

Browse files
committed
improve output
1 parent 32a64d3 commit ea2443b

2 files changed

Lines changed: 10 additions & 5 deletions

File tree

loda/ml/keras/program_generation_rnn.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,8 @@ def summary(self, line_length=None, positions=None, print_fn=None,
9595
print("Sample size:", self.sample_size)
9696
print("Trained samples:", self.num_samples)
9797
print("Trained programs:", len(self.program_ids))
98+
print("Operation per sample:", self.num_ops_per_sample)
99+
print("Nop separators:", self.num_nops_separator)
98100

99101
@classmethod
100102
def from_config(cls, config):
@@ -237,8 +239,8 @@ def get_stats_info_str(self) -> str:
237239
generated programs: 233, speed: 17.43 programs/s, token errors: 0.03%, program errors: 6.01%, separator overhead: -0.40%
238240
```
239241
"""
240-
separator_overhead = 1 - (self.num_generated_nops /
241-
(self.num_generated_programs * self.model.num_nops_separator))
242+
separator_overhead = (
243+
self.num_generated_nops / (self.num_generated_programs * self.model.num_nops_separator)) - 1
242244
return "generated programs: {}, speed: {:.2f} programs/s, token errors: {:.2f}%, program errors: {:.2f}%, separator overhead: {:.2f}%".format(
243245
self.num_generated_programs,
244246
self.num_generated_programs / (time.time() - self.start_time),

mine.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import datetime
22
import os.path
33

4+
from loda.lang import Operation, Program
45
from loda.oeis import ProgramCache, Sequence
56
from loda.runtime import Interpreter
67
from loda.mine import Miner
@@ -11,7 +12,9 @@ def mine(model_path: str):
1112

1213
model = load_model(model_path)
1314
model.summary()
14-
generator = Generator(model, num_lanes=10)
15+
initial_program = Program()
16+
initial_program.operations.append(Operation("mov $1,1"))
17+
generator = Generator(model, initial_program=initial_program, num_lanes=10)
1518
programs_dir = os.path.expanduser("~/loda/programs/oeis")
1619
program_cache = ProgramCache(programs_dir)
1720
ignore_ids = set()
@@ -28,10 +31,10 @@ def mine(model_path: str):
2831
while True:
2932
miner()
3033
i += 1
31-
if i % 100 == 0:
34+
if i % 10 == 0:
3235
ct = datetime.datetime.now()
3336
print(ct, generator.get_stats_info_str())
3437

3538

3639
if __name__ == "__main__":
37-
mine(os.path.expanduser("~/scripts/model-001"))
40+
mine(os.path.expanduser("~/scripts/model-075"))

0 commit comments

Comments
 (0)