predict_NVU: predict_NVU

In Erickcufe/textCells: What the Package Does (One Line, Title Case)

predict_NVU

Description

This function predicts the cell type of a given gene in the Neurovascular Unit (NVU) using a pre-trained XGBoost model.

Usage

predict_NVU(gene)

Arguments

gene	A string representing the gene for which the cell type is being predicted.

Details

The function loads a pre-trained XGBoost model from the file 'XGboost_allattr.rds'. The function then retrieves the attributes for the gene using the Medline database and calculates the TF-IDF scores. The XGBoost model is then used to predict the cell type based on these attributes. The function filters out predictions with a probability of less than 0.55. The average prediction scores for each cell type are calculated and returned in a data frame.

Value

A data frame containing the predicted cell type scores for the gene.

Examples

## Not run: 
  predict_NVU("ABCA1")

## End(Not run)

Erickcufe/textCells documentation built on May 20, 2023, 11:45 p.m.