[UI] Added option to parametrize the min probability for creating high level tags

This commit is contained in:
emeric
2016-05-30 13:57:18 +02:00
parent 445f459654
commit fd9fdb8e87
5 changed files with 55 additions and 8 deletions
+3
View File
@@ -126,6 +126,9 @@ Handler::Handler(Wt::Dbo::SqlConnectionPool& connectionPool)
if (!Setting::exists(_session, "tags_highlevel_acousticbrainz"))
Setting::setBool(_session, "tags_highlevel_acousticbrainz", true);
if (!Setting::exists(_session, "tags_highlevel_acousticbrainz_min_probability"))
Setting::setInt(_session, "tags_highlevel_acousticbrainz_min_probability", 90);
if (!Setting::exists(_session, "tags_similarity_acousticbrainz"))
Setting::setBool(_session, "tags_similarity_acousticbrain", false);
}
@@ -35,7 +35,7 @@ static Cluster::pointer getCluster(std::string type, std::string value)
return cluster;
}
static std::list<std::string> getClustersFromFeature(Feature::Type& feature)
static std::list<std::string> getClustersFromFeature(Feature::Type& feature, double minProb)
{
struct HighLevelNodeDesc
{
@@ -168,6 +168,7 @@ void
HighLevelCluster::processDatabaseUpdate(Updater::Stats stats)
{
bool createTags = Setting::getBool(UpdaterDboSession(), "tags_highlevel_acousticbrainz", false);
double minProb = Setting::getInt(UpdaterDboSession(), "tags_highlevel_acousticbrainz_min_probability", false) / 100.;
LMS_LOG(DBUPDATER, INFO) << "Creating high level based clusters...";
@@ -187,7 +188,7 @@ HighLevelCluster::processDatabaseUpdate(Updater::Stats stats)
if (!Feature::Store::instance().get(UpdaterDboSession(), trackId, "high_level", feature))
continue;
newClusterNames = getClustersFromFeature(feature);
newClusterNames = getClustersFromFeature(feature, minProb);
}
Wt::Dbo::Transaction transaction(UpdaterDboSession());