text (version 0.9.50)

textWordPrediction: Compute predictions based on single words for plotting words. The word embeddings of single words are trained to predict the mean value associated with that word. P-values does NOT work yet.

Description

Compute predictions based on single words for plotting words. The word embeddings of single words are trained to predict the mean value associated with that word. P-values does NOT work yet.

Usage

textWordPrediction(
  words,
  single_word_embeddings = single_word_embeddings_df,
  x,
  y = NULL,
  seed = 1003,
  case_insensitive = TRUE,
  ...
)

Arguments

words

Word or text variable to be plotted.

single_word_embeddings

Word embeddings from textEmbed for individual words (i.e., decontextualized embeddings).

x

Numeric variable that the words should be plotted according to on the x-axes.

y

Numeric variable that the words should be plotted according to on the y-axes (y=NULL).

seed

Set different seed.

case_insensitive

When TRUE all words are made lower case.

...

Training options from textTrainRegression().

Value

A dataframe with variables (e.g., including trained (out of sample) predictions, frequencies, p-values) for the individual words that is used for the plotting in the textProjectionPlot function.

Examples

# NOT RUN {
# Data
# Pre-processing data for plotting
# }
# NOT RUN {
df_for_plotting <- textWordPrediction(
  words = Language_based_assessment_data_8$harmonywords,
  single_word_embeddings = word_embeddings_4$singlewords_we,
  x = Language_based_assessment_data_8$hilstotal
)
df_for_plotting
# }
# NOT RUN {
#' @seealso see \code{\link{textProjection}}
# }