2 Evaluation using Trio
Source:vignettes/v02_Evaluation_using_Trio.Rmd
v02_Evaluation_using_Trio.Rmd
Import microbiome data
The Trio
object in BenchHub
can take
datasets provided by users. To demonstrate, its ability to take
user-provided datasets, we’ll be using a microbiome dataset called
Lubomski
obtained from the PD16Sdata
package.
The following code will import the Lubomksi
data into R.
lubomski_microbiome_data.Rdata
contains two data objects:
x
and lubomPD
. x
is a 575 by 1192
matrix containing the abundance of 1192 microbial taxa for 575 samples.
lubom_pd
is a factor vector of binary patient classes for
575 samples where where 1
represents PD
and
0
represents HC
.
# import the microbiome data
data("lubomski_microbiome_data", package = "BenchHub")
# check the dimension of the microbiome matrix
dim(x)
## [1] 575 1192
# check the length of the patient status
length(lubomPD)
## [1] 575
The task we’ll be evaluating uses a binary classification task where
each sample is either a Parkinson’s Disease (PD) patient or Healthy
Control (HC). Once the data are ready to be inputted to
Trio
, we can load Trio
.
Initialise a Trio object
To initialise a Trio object, we use a new()
method.
trio <- Trio$new(data = x, evidence = list(Diagnosis = list(evidence = lubomPD, metrics = "Balanced Accuracy")),
metrics = list(`Balanced Accuracy` = balAccMetric),
datasetID = "lubomski_microbiome")
## Warning: No sample IDs found on evidence. Assuming same order as data
## and adding them.
The supporting evidence is the disease diagnosis, which is the
lubomPD
vector that we extracted from the
Lubomski
data above and the relevant metric to evaluate it
is Balanced Accuracy. Next, we will use this microbiome data to
illustrate two examples, 1) evaluation with cross validation, 2)
evaluation without cross validation.
Cross validation - using the trio split function
In this example, we show a typical classification scenario to classify the patient status, where we need to split the data into training and testing set.
Step 1: Obtain relevant data for the model. To build a model, the
following code will extract the data matrix and patient status outcome
from Trio
object.
# get the gold standard from the Trio object
x <- trio$data
y <- trio$getEvidence("Diagnosis")
Step 2: For repeated cross-validation, the split()
function will split the y
vector (i.e.,
evidence
) into the number of folds and repeats users want.
In this case we used n_fold = 2
and
n_repeat = 5
(i.e., 2-fold cross-validation with 10
repeats). Then users can get cross-validation indices
(cv_ind
) for each sample, through
splitIndices
. This gives a simple list where each element
represents a combination of folds and repeats for each sample.
# get train and test indices
trio$split(y = y, n_fold = 2, n_repeat = 5)
CVindices <- trio$splitIndices
Step 5: Build the classification model and evaluate. We can now use a for loop to cross-validate evaluation results.
Using the cv_ind
, user can subset to training and test
data. As an example, we build a LASSO regression model as a
classification model on the training data, then make predictions on the
test data.
Once we get the prediction, we pass the pred
vector to
Trio
and ref to the name of the evidence
(patient_status
= pred
). This tells the
evaluate()
function to compare the pred
vector
with the supporting evidence patient_status
stored in the
Trio
object. It will then compute the balanced accuracy,
because this is the metric we have specified.
set.seed(1234)
# loop through the 2 folds x 5 repeats = 10 runs
result <- do.call(rbind, mapply(function(trainIDs, crossValID)
{
x_train <- x[trainIDs, ]
x_test <- x[-trainIDs, ]
y_train <- y[trainIDs]
y_test <- y[-trainIDs]
# find the best lambda for LASSO regression
cv_lasso <- cv.glmnet(as.matrix(x_train), y_train, alpha = 1, family = "binomial")
lam <- cv_lasso$lambda.1se
# fit a model with the best lambda on training data
fit <- glmnet(x_train, y_train, alpha = 1, lambda = lam, family = "binomial")
# evaluate the model on test data
pred <- predict(fit, x_test, s = "lambda.min", type = "class")
pred <- setNames(as.factor(as.vector(pred)), rownames(pred))
# get the chosen evaluation metric from the Trio
eval_res <- trio$evaluate(list(lasso = list(Diagnosis = pred)))
# keep track of the repeat and fold information
eval_res$track <- crossValID
eval_res
}, CVindices, names(CVindices), SIMPLIFY = FALSE))
After cross-validation, we can visualise cross-validation results by averaging results across folds within each repeats.
result$fold <- unlist(lapply(strsplit(result$track, ".", fixed = TRUE), `[`, 1))
result$repeats <- unlist(lapply(strsplit(result$track, ".", fixed = TRUE), `[`, 2))
result <- result %>%
dplyr::group_by(datasetID, method, evidence, metric, repeats) %>%
dplyr::summarize(result = mean(result))
## `summarise()` has grouped output by 'datasetID', 'method', 'evidence',
## 'metric'. You can override using the `.groups` argument.
# visualise the result
boxplot(result$result)
Without Cross-validation
In this example, we show a simpler case without cross validation, where we have a prediction that we want to compare with the ground truth directly. For example, we may have simulated a single-cell count matrix and want to compare with an experimental count matrix on various attributes.
Here, we use the microbiome dataset to demonstrate the above case.
Step 1: Simulate a matrix of the same size as the microbiome data using negative binomial. Step 2: Define a function that calculate the sparsity of a given matrix. Step 3: Define a metric that compare difference of two values. Step 4: Evaluate the difference in sparsity of the microbiome data and the simulated data.
set.seed(1)
# generate a simulated matrix
sim <- rnbinom(nrow(x) * ncol(x), size = 1, mu = 1)
sim <- Matrix(sim, nrow = nrow(x), ncol = ncol(x))
# function that calculate the sparsity of a matrix
calc_sparsity <- function(data) {
sparsity <- sum(data == 0) / length(data)
return(sparsity)
}
# metric that compare the difference of two values
calc_diff <- function(evidence, predicted) {
return(predicted - evidence)
}
# add metric that we just defined
trio$addMetric(name = "Difference", metric = calc_diff)
# calculate the sparsity of the data, and then input into the supporting evidence
trio$addEvidence(name = "Sparsity", evidence = calc_sparsity, metrics = "Difference")
# because we used negative binomial to simulate a matrix
# we will name the method as "negative binomial"
# then calculate the sparsity of the simulated matrix and compare with the Sparsity supporting evidence
eval_res <- trio$evaluate(
list(negative_binomial = list(Sparsity = sim))
)
eval_res
## # A tibble: 1 × 5
## datasetID method evidence metric result
## <chr> <chr> <chr> <chr> <dbl>
## 1 lubomski_microbiome negative_binomial Sparsity Difference -0.323
From the evaluation result, we see there is a -0.32 difference in the sparsity between the microbiome data and the data simulated with negative binomial.
Importing result to BenchmarkInsights
The result dataframe obtained from Trio
evaluation can
be passed to BenchmarkInsights
for subsequent
visualisation. BenchmarkInsights
is another structure in
BenchHub
that provide a list of functions to analyse and
visualise the benchmarking results from multiple perspectives.
Note, to make the result dataframe compatible for BenchmarkInsights, please make sure the dataframe is in the format shown below, where the column names are “datasetID”, “method”, “evidence”, “metric” , “result”, and that there is one row for each result.
# in the with cross validation result, we need to average the results from multiple repeats to give one value
result <- result %>%
dplyr::group_by(datasetID, method, evidence, metric) %>%
dplyr::summarize(result = mean(result))
## `summarise()` has grouped output by 'datasetID', 'method', 'evidence'. You can
## override using the `.groups` argument.
result <- rbind(result, eval_res)
result
## # A tibble: 2 × 5
## # Groups: datasetID, method, evidence [2]
## datasetID method evidence metric result
## <chr> <chr> <chr> <chr> <dbl>
## 1 lubomski_microbiome lasso Diagnosis Balanced Accuracy 0.682
## 2 lubomski_microbiome negative_binomial Sparsity Difference -0.323
Please see Vignette 3 for the details on the visualisations and
functions in BenchmarkInsights
.
Session Info
## R version 4.5.1 (2025-06-13)
## Platform: x86_64-pc-linux-gnu
## Running under: Ubuntu 24.04.3 LTS
##
## Matrix products: default
## BLAS: /usr/lib/x86_64-linux-gnu/openblas-pthread/libblas.so.3
## LAPACK: /usr/lib/x86_64-linux-gnu/openblas-pthread/libopenblasp-r0.3.26.so; LAPACK version 3.12.0
##
## locale:
## [1] LC_CTYPE=C.UTF-8 LC_NUMERIC=C LC_TIME=C.UTF-8
## [4] LC_COLLATE=C.UTF-8 LC_MONETARY=C.UTF-8 LC_MESSAGES=C.UTF-8
## [7] LC_PAPER=C.UTF-8 LC_NAME=C LC_ADDRESS=C
## [10] LC_TELEPHONE=C LC_MEASUREMENT=C.UTF-8 LC_IDENTIFICATION=C
##
## time zone: UTC
## tzcode source: system (glibc)
##
## attached base packages:
## [1] stats graphics grDevices utils datasets methods base
##
## other attached packages:
## [1] glmnet_4.1-10 Matrix_1.7-3 lubridate_1.9.4 forcats_1.0.0
## [5] stringr_1.5.1 dplyr_1.1.4 purrr_1.1.0 readr_2.1.5
## [9] tidyr_1.3.1 tibble_3.3.0 ggplot2_3.5.2 tidyverse_2.0.0
## [13] BenchHub_0.99.5 BiocStyle_2.36.0
##
## loaded via a namespace (and not attached):
## [1] gridExtra_2.3 httr2_1.2.1 sandwich_3.1-1
## [4] rlang_1.1.6 magrittr_2.0.3 multcomp_1.4-28
## [7] polspline_1.1.25 compiler_4.5.1 survAUC_1.3-0
## [10] systemfonts_1.2.3 vctrs_0.6.5 reshape2_1.4.4
## [13] quantreg_6.1 shape_1.4.6.1 pkgconfig_2.0.3
## [16] fastmap_1.2.0 backports_1.5.0 utf8_1.2.6
## [19] ggstance_0.3.7 rmarkdown_2.29 tzdb_0.5.0
## [22] ragg_1.4.0 MatrixModels_0.5-4 xfun_0.53
## [25] cachem_1.1.0 jsonlite_2.0.0 broom_1.0.9
## [28] cluster_2.1.8.1 R6_2.6.1 bslib_0.9.0
## [31] stringi_1.8.7 RColorBrewer_1.1-3 rpart_4.1.24
## [34] jquerylib_0.1.4 cellranger_1.1.0 iterators_1.0.14
## [37] Rcpp_1.1.0 bookdown_0.44 knitr_1.50
## [40] zoo_1.8-14 base64enc_0.1-3 parameters_0.28.1
## [43] timechange_0.3.0 splines_4.5.1 nnet_7.3-20
## [46] tidyselect_1.2.1 rstudioapi_0.17.1 yaml_2.3.10
## [49] codetools_0.2-20 curl_7.0.0 lattice_0.22-7
## [52] plyr_1.8.9 withr_3.0.2 bayestestR_0.17.0
## [55] evaluate_1.0.5 marginaleffects_0.29.0 foreign_0.8-90
## [58] desc_1.4.3 survival_3.8-3 pillar_1.11.0
## [61] BiocManager_1.30.26 foreach_1.5.2 checkmate_2.3.3
## [64] insight_1.4.1 generics_0.1.4 hms_1.1.3
## [67] scales_1.4.0 glue_1.8.0 rms_8.0-0
## [70] Hmisc_5.2-3 tools_4.5.1 data.table_1.17.8
## [73] SparseM_1.84-2 fs_1.6.6 mvtnorm_1.3-3
## [76] grid_4.5.1 datawizard_1.2.0 colorspace_2.1-1
## [79] nlme_3.1-168 googlesheets4_1.1.1 patchwork_1.3.2
## [82] performance_0.15.1 htmlTable_2.4.3 googledrive_2.1.1
## [85] splitTools_1.0.1 Formula_1.2-5 cli_3.6.5
## [88] rappdirs_0.3.3 textshaping_1.0.2 gargle_1.5.2
## [91] gtable_0.3.6 ggcorrplot_0.1.4.1 ggsci_3.2.0
## [94] sass_0.4.10 digest_0.6.37 ggrepel_0.9.6
## [97] TH.data_1.1-3 htmlwidgets_1.6.4 farver_2.1.2
## [100] htmltools_0.5.8.1 pkgdown_2.1.3 lifecycle_1.0.4
## [103] dotwhisker_0.8.4 MASS_7.3-65