def test_io_static():
host = '127.0.0.1'
user = '******'
password = '******'
alg = 'iforest'
rng = np.random.RandomState(42)
np.random.seed(42)
conn, cursor = connect_server(host, user, password)
ground_truth_whole = insert_demo_data(conn,
cursor,
'db',
't',
ground_truth_flag=True)
data, ground_truth = query_data(conn,
cursor,
'db',
't',
time_serie_name='ts',
ground_truth=ground_truth_whole,
start_time='2019-07-20 00:00:00',
end_time='2019-08-20 00:00:00',
time_serie=False,
ground_truth_flag=True)
clf = algorithm_selection(alg, random_state=rng, contamination=0.1)
print('Start processing:')
start_time = time.clock()
clf.fit(data)
prediction_result = clf.predict(data)
outlierness = clf.decision_function(data)
output_performance(alg, ground_truth, prediction_result,
time.clock() - start_time, outlierness)
visualize_distribution_static(data, prediction_result, outlierness)
visualize_distribution(data, prediction_result, outlierness)
visualize_outlierscore(outlierness, prediction_result, contamination=0.1)
data, ground_truth = query_data(conn,
cursor,
'db',
't',
time_serie_name='ts',
ground_truth=ground_truth_whole,
start_time=None,
end_time='2019-08-20 00:00:00',
time_serie=False,
ground_truth_flag=True)
data, ground_truth = query_data(conn,
cursor,
'db',
't',
time_serie_name='ts',
ground_truth=ground_truth_whole,
start_time='2019-07-20 00:00:00',
end_time=None,
time_serie=False,
ground_truth_flag=True)
clf = alg_selector.model_selector(max_evals=2)
else:
clf = algorithm_selection(args.algorithm,
random_state=rng,
contamination=args.contamination)
start_time = time.clock()
clf.fit(data)
prediction_result = clf.predict(data)
outlierness = clf.decision_function(data)
anomaly_scores = clf.anomaly_likelihood(data)
if args.ground_truth:
output_performance(args.algorithm, ground_truth, prediction_result,
time.clock() - start_time, anomaly_scores)
if args.visualize_distribution and args.ground_truth:
if not args.time_stamp:
visualize_distribution_static(data, prediction_result, outlierness,
args.saving_path)
visualize_distribution(data, prediction_result, outlierness,
args.saving_path)
visualize_outlierscore(outlierness, prediction_result,
args.contamination, args.saving_path)
else:
visualize_distribution_time_serie(clf.ts, data, args.saving_path)
visualize_outlierscore(outlierness, prediction_result,
args.contamination, args.saving_path)
conn.close()