blob: ab9dba6b7c9a2de5b9f6b789d87143847f7522dd (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
#' Subset an idframe object
#'
#' Subsetting method for objects of class \code{idframe}
#'
#' @param object an object of class \code{idframe}
#' @param indices the indices that need to be subsetted
#' @export
dataSlice <- function(object,indices){
# check if the class is correct
if(class(object)!='idframe')
stop("Not an idframe object")
if(!all(indices %in% seq(to=dim(object$output)[1],by=1)))
stop("Invalid indices")
trim <- object
trim$output <- trim$output[indices,]
trim$input <- trim$input[indices,]
if(trim$type=="freq"){
trim$frequncies <- trim$frequencies[indices]
} else {
trim$t.start <- trim$t.start + trim$Ts*(indices[1]-1)
trim$t.end <- trim$t.start + trim$Ts*(length(indices)-1)
}
return(trim)
}
#' Split data into training and validation sets
#'
#' The function splits the data into training and validation sets and returns them bundled
#' as a list. The size of the sets are determined by the parameter \code{p}.
#'
#' @param object an object of class \code{idframe}
#' @param p the percentage of the data that goes to training (Default : \code{0.6})
#' @return list containing estimation and validation idframe objects
#' @export
dataPartition <- function(object,p=0.6){
# check if the class is correct
if(class(object)!='idframe')
stop("Not an idframe object")
index <- seq_along(object$output[,1])
trainIndex <- index[1:round(p*length(index))]
testIndex <- index[!(index %in% trainIndex)]
train <- dataSlice(object,trainIndex)
test <- dataSlice(object,testIndex)
return(list(estimation=train,validation=test))
}
|
