forked from biocuckooHXH/pFunK
-
Notifications
You must be signed in to change notification settings - Fork 0
/
predict.py
42 lines (36 loc) · 1.22 KB
/
predict.py
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
"""
Filename: predict.py
Author: yellower
"""
from torch.utils.data import DataLoader
import pandas as pd
from keras.models import load_model
import numpy as np
from keras import backend as K
from tqdm import tqdm
from utils import to_categorical,caculate_auc
from sklearn.model_selection import StratifiedKFold
from keras.callbacks import ModelCheckpoint
import matplotlib.pyplot as plt
from tensorflow import keras
from sklearn.model_selection import train_test_split
import time
import tensorflow as tf
# tf.enable_eager_execution()
from tensorflow import keras
from tensorflow.keras import layers
from keras.models import Model
from keras.layers import *
import tensorflow as tf
import numpy as np
if __name__ == '__main__':
indentified_CPLM_all_K_concat = pd.read_table('path_for_your_sites.txt', sep='\t', index_col=0)
model = load_model('path_for_your_model')
fasta_K_emb_list = []
for item in indentified_CPLM_all_K_concat['fasta_K']:
fasta_K_emb_list.append([amino_acid_dict[i] for i in item])
if len(item) != 61:
print(len(item))
fasta_K_emb_arrays = np.array(fasta_K_emb_list)
score = model.predict(fasta_K_emb_arrays)
print(score)