diff --git a/fastlib/u/wguan/nnsvm/nnsvm.h b/fastlib/u/wguan/nnsvm/nnsvm.h index 08ac3e0ccc..cb8e1a797b 100644 --- a/fastlib/u/wguan/nnsvm/nnsvm.h +++ b/fastlib/u/wguan/nnsvm/nnsvm.h @@ -223,8 +223,10 @@ void NNSVM::SaveModel(String modelfilename) { // TODO: use XML template void NNSVM::LoadModel(Dataset* testset, String modelfilename) { - /* Init */ + /* Init */ + //fprintf(stderr, "modelfilename= %s\n", modelfilename.c_str()); num_features_ = testset->n_features() - 1; + model_.w_.Init(num_features_); /* load model file */ FILE *fp = fopen(modelfilename, "r"); @@ -258,11 +260,12 @@ void NNSVM::LoadModel(Dataset* testset, String modelfilename) { for (index_t s= 0; s < num_features_; s++) { fscanf(fp,"%lf",&temp_f); model_.w_[s] = temp_f; - } + } + break; } } support_vectors_.Init(num_features_, model_.num_sv_); - model_.sv_coef_.Init(model_.num_sv_); + model_.sv_coef_.Init(model_.num_sv_); while (1) { fscanf(fp,"%80s",cmd); if (strcmp(cmd, "svs")==0) { @@ -273,7 +276,8 @@ void NNSVM::LoadModel(Dataset* testset, String modelfilename) { fscanf(fp,"%lf",&temp_f); support_vectors_.set(j, i, temp_f); } - } + } + break; } } fclose(fp); @@ -335,7 +339,7 @@ void NNSVM::BatchClassify(Dataset* testset, String testlablefilename) { */ template void NNSVM::LoadModelBatchClassify(Dataset* testset, String modelfilename, String testlabelfilename) { - LoadModel(testset, modelfilename); + LoadModel(testset, modelfilename); BatchClassify(testset, testlabelfilename); }