WIF - Classifiers - Introduce method for ML model reload for ScikitMLClassifier

TG-45

Author: plnyrich

include/wif/classifiers/scikitMlClassifier.hpp

@@ -43,6 +43,11 @@ class ScikitMlClassifier : public Classifier {
 	 */
 	void setFeatureSourceIDs(const std::vector<FeatureID>& sourceFeatureIDs) override;
 
+	/**
+	 * @brief Reload used ML model from disk
+	 */
+	void reloadModelFromDisk();
+
 	/**
 	 * @brief Classify single flowFeature object
 	 *

src/wif/classifiers/scikitMlClassifier.cpp

@@ -28,6 +28,11 @@ void ScikitMlClassifier::setFeatureSourceIDs(const std::vector<FeatureID>& sourc
 	m_scikitMlWrapper->setFeatureSourceIDs(sourceFeatureIDs);
 }
 
+void ScikitMlClassifier::reloadModelFromDisk()
+{
+	m_scikitMlWrapper->reloadModel();
+}
+
 ClfResult ScikitMlClassifier::classify(const FlowFeatures& flowFeatures)
 {
 	return m_scikitMlWrapper->classify({flowFeatures})[0];