这是我想调用的infer.py中的函数:
def _predict_ids_to_seq(predict_ids, id2word, beam_size):
predicts = []
for single_predict in predict_ids:
for i in range(beam_size):
predict_list = np.ndarray.tolist(single_predict[:, :, i]) #一维全取,二维全取,三维取i值
predict_seq = [id2word[idx] for idx in predict_list[0]
if idx in id2word if idx >=4]
predicts.append("".join(predict_seq))
return predicts
我想将predicts列表赋给另一个py文件中的变量probs_temp_int
我是这样调用的:
from seq2seq_dialog.infer import _predict_ids_to_seq
...
probs_temp_int = _predict_ids_to_seq(predict_ids, id2word, beam_size)
但是由于 _predict_ids_to_seq函数的三个变量还需其他计算才能得到,所以不能给定值,求助如何调用这种函数的返回值?