View source: R/as_prediction_clust.R
as_prediction_clust | R Documentation |
Convert object to a PredictionClust.
as_prediction_clust(x, ...)
## S3 method for class 'PredictionClust'
as_prediction_clust(x, ...)
## S3 method for class 'data.frame'
as_prediction_clust(x, ...)
x |
(any) |
... |
(any) |
PredictionClust.
if (requireNamespace("e1071")) {
# create a prediction object
task = tsk("usarrests")
learner = lrn("clust.kmeans")
learner = lrn("clust.cmeans", predict_type = "prob")
learner$train(task)
p = learner$predict(task)
# convert to a data.table
tab = as.data.table(p)
# convert back to a Prediction
as_prediction_clust(tab)
# split data.table into a 3 data.tables based on UrbanPop
f = cut(task$data(rows = tab$row_ids)$UrbanPop, 3)
tabs = split(tab, f)
# convert back to list of predictions
preds = lapply(tabs, as_prediction_clust)
# calculate performance in each group
sapply(preds, function(p) p$score(task = task))
}
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.