File size: 1,559 Bytes
bf928ee | 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 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | import os
import pickle
from _bootstrap import DATA_DIR, WEIGHT_DIR
from model import get_model
from sklearn.metrics import roc_auc_score,average_precision_score
cdr_kmer = 3
ag_kmer = 1
data_path = DATA_DIR / 'features' / ('cdr_kmer' + str(cdr_kmer) + '_ag_kmer' + str(ag_kmer))
weight_path = WEIGHT_DIR / ('cdr_kmer' + str(cdr_kmer) + '_ag_kmer' + str(ag_kmer)) / 'Model99.h5'
with (data_path / 'cdr_features_te.pickle').open('rb') as binary_reader:
cdr_features_te = pickle.load(binary_reader)
with (data_path / 'ag_features_te.pickle').open('rb') as binary_reader:
ag_features_te = pickle.load(binary_reader)
# Test data
dtest_cdr_ids = []
dtest_cdr_number_ids = []
dtest_ag_ids = []
dtest_labels = []
dtest_labels_pos = 0
dtest_labels_neg = 0
for feature in cdr_features_te:
dtest_cdr_ids.append(feature.input_ids)
dtest_cdr_number_ids.append(feature.cdr_number_ids)
dtest_labels.append(feature.label_id)
dtest_labels_pos = dtest_labels_pos + feature.label_id
dtest_labels_neg = len(dtest_labels) - dtest_labels_pos
for feature in ag_features_te:
dtest_ag_ids.append(feature.input_ids)
model=None
model=get_model()
model.load_weights(str(weight_path))
print("****************Testing the model ****************")
labels_pred = model.predict([dtest_cdr_ids, dtest_cdr_number_ids, dtest_ag_ids])
auc_test = roc_auc_score(dtest_labels, labels_pred)
aupr_test = average_precision_score(dtest_labels, labels_pred)
print("AUC_test : ", auc_test)
print("AUPR_test : ", aupr_test)
|