weixin_43981428
NHR33
采纳率100%
2020-09-07 09:20 阅读 227

如何调用其他py文件函数的返回值?

这是我想调用的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函数的三个变量还需其他计算才能得到,所以不能给定值,求助如何调用这种函数的返回值?

  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 复制链接分享

1条回答 默认 最新

相关推荐