Skip to main content

This package is used to predict intimacy for questions

Project description



question-intimacy is a package used to estimate the intimacy of questions. It is released with EMNLP 2020 paper Quantifying Intimacy in Language.


Use pip

If pip is installed, question-intimacy could be installed directly from it:

pip3 install question-intimacy


transformers >= 3.1.0

Usage and Example

Notes: During your first usage, the package will download a model file automatically, which is about 500MB.

Construct the Predictor Object

>>> from question_intimacy.predict_intimacy import IntimacyEstimator
>>> inti = IntimacyEstimator()

Cuda is disabled by default, to allow GPU calculation, please use

>>> from question_intimacy.predict_intimacy import IntimacyEstimator
>>> inti = IntimacyEstimator(cuda=True)


predict is the core method of this package, which takes a single text of a list of texts, and returns a list of raw values in [-1,1] (higher means more intimate, while lower means less).

# Predict intimacy for one question
>>> text = 'What is this movie ?''
>>> inti.predict(text,type='list')

# 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]


Jiaxin Pei (

Project details

Release history Release notifications | RSS feed

This version


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

question_intimacy-1.1.tar.gz (2.9 kB view hashes)

Uploaded source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page