Learn R Programming

mildsvm (version 0.4.1)

predict.svor_exc: Predict method for svor_exc object

Description

Predict method for svor_exc object

Usage

# S3 method for svor_exc
predict(
  object,
  new_data,
  type = c("class", "raw"),
  layer = c("instance", "bag"),
  new_bags = "bag_name",
  ...
)

Value

A tibble with nrow(new_data) rows. If type = 'class', the tibble will have a column .pred_class. If type = 'raw', the tibble will have a column .pred.

Arguments

object

An object of class svor_exc.

new_data

A data frame to predict from. This needs to have all of the features that the data was originally fitted with.

type

If 'class', return predicted values with threshold of 0 as -1 or +1. If 'raw', return the raw predicted scores.

layer

If 'bag', return predictions at the bag level. If 'instance', return predictions at the instance level.

new_bags

A character or character vector. Can specify a singular character that provides the column name for the bag names in new_data (default 'bag_name'). Can also specify a vector of length nrow(new_data) that has bag name for each row.

...

Arguments passed to or from other methods.

Author

Sean Kent

Details

When the object was fitted using the formula method, then the parameter new_bags is not necessary, as long as the names match the original function call.

See Also

svor_exc() for fitting the svor_exc object.

Examples

Run this code
data("ordmvnorm")
x <- ordmvnorm[, 3:7]
y <- attr(ordmvnorm, "instance_label")

mdl1 <- svor_exc(x, y)
predict(mdl1, x)
predict(mdl1, x, type = "raw")

Run the code above in your browser using DataLab