Commit a21cbff4 authored by wangys_biolab's avatar wangys_biolab

Update SSA_embedding.py

parent 901c9a57
......@@ -124,7 +124,7 @@ def SSA_Embed(input_seq):
final_only = True
for key, value in input_seq.items():
PID_.append(key)
sequence = value.encode("utf-8")
sequence = str(value).encode("utf-8")
z = embed_sequence(sequence, lm_embed, lstm_stack, proj
, include_lm=include_lm, final_only=final_only
, pool='avg', use_cuda=True)
......@@ -156,7 +156,7 @@ def BiLSTM_Embed(input_seq):
for key,value in input_seq.items():
PID.append(key)
sequence = value.encode("utf-8")
sequence = str(value).encode("utf-8")
z = embed_sequence(sequence, lm_embed, lstm_stack, proj
, final_only=False,include_lm = True
, pool='avg', use_cuda=True)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment