apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

nnsvm1124-2

Browse Source
This commit is contained in:
wguan
2008-01-24 22:35:13 +00:00
parent 7bd2bdd05d
commit 1069dbedec
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastlib/u/wguan/nnsvm/nnsvm.h
+9 -5
View File
@@ -223,8 +223,10 @@ void NNSVM<TKernel>::SaveModel(String modelfilename) {
// TODO: use XML
template<typename TKernel>
void NNSVM<TKernel>::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<TKernel>::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<TKernel>::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<TKernel>::BatchClassify(Dataset* testset, String testlablefilename) {
*/
template<typename TKernel>
void NNSVM<TKernel>::LoadModelBatchClassify(Dataset* testset, String modelfilename, String testlabelfilename) {
LoadModel(testset, modelfilename);
LoadModel(testset, modelfilename);
BatchClassify(testset, testlabelfilename);
}