plot_heatmap: Coefficient Heatmap for orthoMTL

In orthoMTL: Multi-Task Learning with Orthogonal Constraints

Coefficient Heatmap for orthoMTL

Description

Displays the coefficient matrix as a heatmap. Features (rows) can optionally be reordered by their mean coefficient across tasks.

Usage

plot_heatmap(x, reorder = TRUE)

Arguments

x	Either a fitted "orthoMTL" object or a numeric coefficient matrix with dimensions p x numTasks. If an "orthoMTL" object, coefficients are extracted via coef().
reorder	Logical. If TRUE (default), rows are sorted by mean coefficient across tasks (ascending).

Value

A ggplot2 object.

See Also

coef.orthoMTL, plot_correlation

Examples

set.seed(42)
n <- 100; p <- 10; n_tasks <- 3
X <- matrix(rnorm(n * p), n, p)
colnames(X) <- paste0("V", seq_len(p))
Y <- X %*% matrix(rnorm(p * n_tasks), p) + matrix(rnorm(n * n_tasks), n) * 0.1
colnames(Y) <- c("T1", "T2", "T3")
K <- matrix(1, n_tasks, n_tasks); diag(K) <- 0.5
fit <- orthoMTL(X, Y, lambda = 1e-3, K = K)

# From fitted object
plot_heatmap(fit)

# From raw matrix
plot_heatmap(coef(fit), reorder = FALSE)

orthoMTL documentation built on Aug. 23, 2026, 5:10 p.m.