public class DataSplitter
extends java.lang.Object
Constructor and Description |
---|
DataSplitter(SparseMatrix rateMatrix)
Construct a data splitter with data source of a given rate matrix
|
DataSplitter(SparseMatrix rateMatrix,
int kfold)
Construct a data splitter to split a given matrix into kfolds
|
Modifier and Type | Method and Description |
---|---|
SparseMatrix[] |
getKthFold(int k)
Return the k-th fold as test set (testMatrix), making all the others as
train set in rateMatrix.
|
SparseMatrix[] |
getRatio(double ratio)
Split ratings into two parts: (1-ratio) training, (ratio) testing data
|
void |
sample(int numUsers,
int numItems)
generate a random sample of rate matrix with specified number of users
and items
|
public DataSplitter(SparseMatrix rateMatrix, int kfold)
rateMatrix
- data matrixkfold
- number of folds to splitpublic DataSplitter(SparseMatrix rateMatrix)
rateMatrix
- data sourcepublic SparseMatrix[] getRatio(double ratio)
ratio
- the ratio of testing data over all the ratings.public void sample(int numUsers, int numItems) throws java.lang.Exception
numUsers
- number of users, -1 to use all users;numItems
- number of items, -1 to user all items;java.lang.Exception
public SparseMatrix[] getKthFold(int k)
k
- The index for desired fold.