-
Notifications
You must be signed in to change notification settings - Fork 3
Expand file tree
/
Copy pathmc_learn
More file actions
34 lines (32 loc) · 1015 Bytes
/
mc_learn
File metadata and controls
34 lines (32 loc) · 1015 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#!/usr/bin/env python
# -*- coding:utf-8 -*-
import os
import sys
from optparse import OptionParser, OptionValueError
import markovchains
if __name__ == '__main__':
usage = "usage: %prog filename/dirname [options]"
parser = OptionParser(usage)
parser.add_option('-u','--user',action='store',help=u'発言したユーザ')
(options, args) = parser.parse_args()
user = ''
if options.user != None:
user = options.user
fileordir = os.path.join(os.environ['PWD'],sys.argv[1])
files = []
if os.path.isdir(fileordir):
files = os.listdir(fileordir)
for i in xrange(len(files)):
files[i] = os.path.join(fileordir,files[i])
elif os.path.isfile(fileordir):
files.append(fileordir)
else:
quit()
m = markovchains.MarkovChains()
sentences = []
for file in files:
print file
f = open(file).read().splitlines()
for line in f:
last = m.analyze_sentence(line,user)
m.register_data()