nimare.annotate.text.generate_counts

generate_counts(text_df, text_column='abstract', tfidf=True, min_df=50, max_df=0.5)[source]

Generate tf-idf weights for unigrams/bigrams derived from textual data.

Parameters:

text_df ((D x 2) pandas.DataFrame) – A DataFrame with two columns (‘id’ and ‘text’). D = document.

Returns:

weights_df – A DataFrame where the index is ‘id’ and the columns are the unigrams/bigrams derived from the data. D = document. T = term.

Return type:

(D x T) pandas.DataFrame