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