def _test_table(request, traina, trainb, testa, testb):
con = sqlite3.connect(settings['USERS_DB'].format(session['username'])).cursor()
traininga = con.execute("SELECT plain_text FROM {}".format(testa)).fetchall()
trainingb = con.execute("SELECT plain_text FROM {}".format(testb)).fetchall()
a, b = [], []
for item in traininga:
a.append(item[0])
for item in trainingb:
b.append(item[0])
my_list = [(item, 0) for item in a]
my_list2 = [(item, 1) for item in b]
my_list.extend(my_list2)
random.shuffle(my_list)
train1 = []
train2 = []
for item in my_list:
train1.append(item[0])
train2.append(item[1])
x = FileController()
file = x.fetch_path(session['username'])
f = open(file + "/{}_{}.pickle".format(traina, trainb), 'rb')
model = pickle.load(f)
f.close()
testcase = TestModel(model, train1, train2)
print(testcase['plain_text'])
print(testcase['guess'])
print(testcase['answer'])
print(testcase['percent'])
return render_template('User/results.html', training=testcase)
def _delete_model(request, model):
items = model.replace(' VS. ', '_')
User(session['username']).delete_model(model)
session['models'] = User(session['username']).fetch_models()
x = FileController()
file = x.fetch_path(session['username'])
os.remove(file+"/{}.pickle".format(items))
return redirect("/training")
def _train_table(request, traina, trainb):
con = sqlite3.connect(settings['USERS_DB'].format(session['username'])).cursor()
traininga = con.execute("SELECT plain_text FROM {}".format(traina)).fetchall()
trainingb = con.execute("SELECT plain_text FROM {}".format(trainb)).fetchall()
a, b = [], []
for item in traininga:
a.append(item[0])
for item in trainingb:
b.append(item[0])
my_list = [(item, 0) for item in a]
my_list2 = [(item, 1) for item in b]
my_list.extend(my_list2)
random.shuffle(my_list)
train1 = []
train2 = []
for item in my_list:
train1.append(item[0])
train2.append(item[1])
print(traina, trainb)
print(train1, train2)
model = MakeModel(train1, train2, [traina, trainb])
x = FileController()
file = x.fetch_path(session['username'])
f = open(file+"/{}_{}.pickle".format(traina, trainb), 'wb')
pickle.dump(model, f)
f.close()
User(session['username']).create_model("{} VS. {}".format(traina, trainb))
session['models'] = User(session['username']).fetch_models()
# f.write('hi')
# f.close()
# print(traina, trainb, testa, testb)
# print("{}/{}v{}".format(x.fetch_path(session['username']), traina, trainb))
# testlist = [(item, 0) for item in c]
# testlist2 = [(item, 1) for item in d]
# testlist.extend(testlist2)
# print(my_list, testlist)
return redirect("/training")
