这个包用于预测问题的亲密度
question-intimac的Python项目详细描述
质疑亲密
简介
问题亲密度是一个用来评估问题亲密度的软件包。它是用
EMNLP 2020论文Quantifying Intimacy in Language
。在
安装
使用pip
如果安装了pip
,则可以直接从中安装问题亲密度:
pip3 install question-intimacy
依赖性
^{pr2}$用法和示例
注意:首次使用时,软件包会自动下载一个约500MB的模型文件。
Construct the Predictor Object
>>> from question_intimacy.predict_intimacy import IntimacyEstimator
>>> inti = IntimacyEstimator()
Cuda默认为禁用,要允许GPU计算,请使用
>>> from question_intimacy.predict_intimacy import IntimacyEstimator
>>> inti = IntimacyEstimator(cuda=True)
predict
predict
是这个包的核心方法,
它接受文本列表中的单个文本,并返回[-1,1]
中的原始值列表(高表示更亲密,而低表示更少)。在
# Predict intimacy for one question
>>> text = 'What is this movie ?''
>>> inti.predict(text,type='list')
-0.2737383
# Predict intimacy for a list of questions (less than a batch)
>>> text = ['What is this movie ?','Why do you hate me ?']
>>> inti.predict(text,type='list')
[-0.2737383, 0.3481976]
# Predict intimacy for a long list of questions
>>> text = [a long list of questions]
>>> inti.predict(text,type='long_list',tqdm=tqdm)
[-0.2737383, 0.3481976, ... ,-0.2737383, 0.3481976]
联系人
裴嘉欣(pedropei@umich.edu)
- 项目
标签: