Commit 76e0bf6b by yahan.li

保存模型前,主动删除旧的模型文件

	修改:     SigerCalculation/RadomForestSAC/RandomForestClassifier.cpp
	修改:     SigerCalculation/package/E3/SigerCalculation
parent 7a69eae5
...@@ -201,8 +201,9 @@ bool RandomForestClassifier::train(const std::vector<std::vector<float>>& X, con ...@@ -201,8 +201,9 @@ bool RandomForestClassifier::train(const std::vector<std::vector<float>>& X, con
try try
{ {
std::string filename = (!m_target.empty()) ? (m_target + "_rf_model.dat") : "rf_model.dat"; std::string filename = (!m_target.empty()) ? (m_target + "_rf_model.dat") : "rf_model.dat";
unlink(filename.c_str());
serialize(filename) << scaler_mean << scaler_std << forest; serialize(filename) << scaler_mean << scaler_std << forest;
SPDLOG_LOGGER_DEBUG(spdlog::get("logger"), "save {} success", filename.c_str()); //SPDLOG_LOGGER_DEBUG(spdlog::get("logger"), "save {} success", filename.c_str());
} }
catch(std::exception &e) { catch(std::exception &e) {
SPDLOG_LOGGER_DEBUG(spdlog::get("logger"), "load model failure: {}", e.what()); SPDLOG_LOGGER_DEBUG(spdlog::get("logger"), "load model failure: {}", e.what());
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment