def getdata_mongo(db_name, file_path):
#backup mongodb of trainbayes
session = MongDBSession()
train_data = session.query_all(db_name)
f = file(file_path, "w")
for i in range(train_data.count()):
for key in train_data[i]:
f.write(key)
f.write("\n")
f.write(str(train_data[i][key]))
f.write("\n")
f.write("\n")
f.close()
def __init__(self):
self.session = MongDBSession()
__author__ = 'wtq'
import numpy
import time
from sklearn import metrics
from detector.logger import AdDetectorLogger
from detector.config import TRAIN_PERMISSION
from permission.predict import AdBernoulliPredict
from permission.predict import AdSVMPredict
from permission.predict import AdRandomForestPredict
from detector.db.session import MongDBSession
from permission.base import BasePermission
from detector.ad.ad_detector_web.ad_detector import ad_detector
base_permission = BasePermission()
session = MongDBSession()
gauss_predict = AdBernoulliPredict()
svm_predict = AdSVMPredict()
random_predict = AdRandomForestPredict()
logger = AdDetectorLogger()
def test_complx_model():
"""
:return:
"""
start_time = time.time()
test_permission = session.query_sort(TRAIN_PERMISSION, 'create', limit=1)
permission_list = test_permission["train-permission"]