diff --git a/test/test_model.py b/test/test_model.py index 3c99447..df125fc 100755 --- a/test/test_model.py +++ b/test/test_model.py @@ -151,17 +151,17 @@ def test_SVM_ILSVRC_S(): def test_THEANO_crop(): - timer.mark() - dilc = ILSVRC.DataILSVRC(base_dir='/data/hadoop/ImageNet/ILSVRC/ILSVRC2013_DET_val', category='Test_crop_pil') - X, Y = dilc.load_data(mode='local', feattype='coef') - timer.report() - X_train, X_test, Y_train, Y_test = cross_validation.train_test_split(X, Y, test_size=0.2, random_state=0) - with open(os.path.join(package_dir,'../res/','ils_crop.pkl'),'wb') as f: - cPickle.dump([(X_train,Y_train),(X_test,Y_test)], f) + # timer.mark() + # dilc = ILSVRC.DataILSVRC(base_dir='/data/hadoop/ImageNet/ILSVRC/ILSVRC2013_DET_val', category='Test_crop_pil') + # X, Y = dilc.load_data(mode='local', feattype='coef') + # timer.report() + # X_train, X_test, Y_train, Y_test = cross_validation.train_test_split(X, Y, test_size=0.2, random_state=0) + # with open(os.path.join(package_dir,'../res/','ils_crop.pkl'),'wb') as f: + # cPickle.dump([(X_train,Y_train),(X_test,Y_test)], f) timer.mark() mtheano = THEANO.ModelTHEANO(toolset='cnn') - mtheano._train_cnn(dataset='../../res/ils_crop.pkl') + mtheano._train_cnn(dataset='../res/ils_crop.pkl') timer.report() -- libgit2 0.21.2