diff options
| -rw-r--r-- | monkey/patterns.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/monkey/patterns.py b/monkey/patterns.py index 8247775..48f9163 100644 --- a/monkey/patterns.py +++ b/monkey/patterns.py @@ -444,7 +444,9 @@ if __name__ == '__main__': split = int(len(users)*0.7) learn_users = set(users[:split]) test_users = set(users[split:]) - print(test_users) + fusers = open('data/users-test-{}.txt'.format(pid), 'wt') + for tu in test_users: + fusers.write('{}\n'.format(tu)) for code, data in submissions.items(): if len(code) > 1000 or prolog_parse(code) is None: |