2017-06-08 14 views
1

私はトレーニングとテストのセットに分けた2つのファイルを持っています。svmfile行列を4つの部分に分割

traindata = load_svmlight_file("training") 
testdata = load_svmlight_file("testing") 
Xtrain = traindata[0].todense() 
ytrain = traindata[1] 
Xtest = testdata[0].todense() 
ytest = testdata[1] 

私は4つの部分に私のXTESTとytestを分割したい:私の本能が使用する

Xtest1、ytest1 Xtest2、ytest2、 Xtest3、ytest3、 Xtest4、ytest4

sklearn.train_test_splitでもわかりません。これを行う最善の方法は何ですか?

答えて

0

あなたはnp.array_splitを使用していることを行うことができます。

import numpy as np 
test=np.arange(21) 
xtest1,xtest2,xtest3,xtest4 = np.array_split(np.random.permutation(test),4) 

を、私はあなたがデータをランダムに分割したい場合にnp.random.permutationを使用。連続した部分に分割したい場合は、np.random.permutationを削除することができます。

関連する問題