File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -53,9 +53,7 @@ ClfResult LightGBMWrapper::classify(const FlowFeatures& flowFeatures)
5353 std::vector<double > dataToClassify; // classified features from flowfeatures are extracted here
5454 int64_t outLen; // length of output result
5555 int numOfClasses; // number of classes
56-
5756 LGBM_BoosterGetNumClasses (m_booster, &numOfClasses);
58-
5957 std::vector<double > pred (numOfClasses); // vector with predictions
6058
6159 for (const auto & featureID : m_featureIDs) {
@@ -78,7 +76,6 @@ ClfResult LightGBMWrapper::classify(const FlowFeatures& flowFeatures)
7876
7977 if (numOfClasses == 1 ) {
8078 double tmp = pred[0 ];
81-
8279 pred.insert (pred.begin (), (1.0 - tmp));
8380 }
8481
@@ -125,7 +122,6 @@ std::vector<ClfResult> LightGBMWrapper::classify(const std::vector<FlowFeatures>
125122
126123 if (numOfClasses == 1 ) {
127124 double tmp = probabilities[0 ];
128-
129125 probabilities.insert (probabilities.begin (), (1.0 - tmp));
130126 }
131127
You can’t perform that action at this time.
0 commit comments