| void Impala::Core::Trec::TrecSVM::LoadModel | ( | std::string | fileName | ) | [inline] |
Definition at line 132 of file TrecSVM.h.
References svm_node::index, mClassLabels, mModel, mNodes, mNrClass, mNrFeatures, mProbs, svm_check_probability_model(), svm_get_labels(), svm_get_nr_class(), and svm_load_model().
Referenced by LoadModel().
00133 { 00134 if ((mModel = svm_load_model(fileName.c_str())) == 0) 00135 std::cout << "TrecSVM: unable to load model " << fileName << std::endl; 00136 if (!svm_check_probability_model(mModel)) 00137 std::cout << "TrecSVM: not a probability model" << std::endl; 00138 00139 // setup svm_node array, assume fixed number of elements 00140 mNodes = new struct svm_node[mNrFeatures+1]; 00141 for (int i=0 ; i<mNrFeatures ; i++) 00142 mNodes[i].index = i+1; 00143 mNodes[mNrFeatures].index = -1; 00144 00145 mNrClass = svm_get_nr_class(mModel); 00146 mClassLabels = new int[mNrClass]; 00147 svm_get_labels(mModel, mClassLabels); 00148 //for (int c=0 ; c<mNrClass ; c++) 00149 // std::cout << "label " << c << " = " << labels[c] << std::endl; 00150 mProbs = new double[mNrClass]; 00151 }
Here is the call graph for this function:
