Skip to content
Snippets Groups Projects
Commit 69ddbac9 authored by Jingfei Hou's avatar Jingfei Hou
Browse files

Update demo.py

parent 9d47b306
No related branches found
No related tags found
No related merge requests found
...@@ -25,3 +25,6 @@ if __name__ == '__main__': ...@@ -25,3 +25,6 @@ if __name__ == '__main__':
model = torch.load('./results/models/'+tar_specie+'.model.pt', map_location='cpu') model = torch.load('./results/models/'+tar_specie+'.model.pt', map_location='cpu')
# 进行评分并返回生成的抗体数据(完整,包括FR和CDR区) # 进行评分并返回生成的抗体数据(完整,包括FR和CDR区)
seq_score(input_cdr, gen_list, model, tar_emb, 'ans_com.csv') seq_score(input_cdr, gen_list, model, tar_emb, 'ans_com.csv')
print("The results of the build can be found in the ans_com file.")
print("The last sequence is the input sequence.")
print("The highest-rated sequence is stored in structure_seq file.")
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment