diff options
author | Andrew Dolgov <[email protected]> | 2015-06-18 10:56:42 +0300 |
---|---|---|
committer | Andrew Dolgov <[email protected]> | 2015-06-18 10:56:42 +0300 |
commit | 5df13f31ef5642e583d9fb8755bd3ed78e16f594 (patch) | |
tree | 01c9ef92f9ff2fa78e84f8c2d2d996d45e4aec45 | |
parent | e0940bfc4fdfad943a111b52569a1917e083b2c0 (diff) |
bayes: fix check for already categorized articles
-rw-r--r-- | plugins/af_sort_bayes/init.php | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/plugins/af_sort_bayes/init.php b/plugins/af_sort_bayes/init.php index d2e659cae..0bdfafa17 100644 --- a/plugins/af_sort_bayes/init.php +++ b/plugins/af_sort_bayes/init.php @@ -263,12 +263,17 @@ class Af_Sort_Bayes extends Plugin { function hook_article_filter($article) { $owner_uid = $article["owner_uid"]; - $nbs = new NaiveBayesianStorage($owner_uid); - $nb = new NaiveBayesianNgram($nbs, 3); + // guid already includes owner_uid so we don't need to include it + $result = $this->dbh->query("SELECT id FROM {$this->sql_prefix}_references WHERE + document_id = '" . $this->dbh->escape_string($article['guid_hashed']) . "'"); - $ref = $nbs->getReference($article["guid"], false); + if (db_num_rows($result) != 0) { + _debug("bayes: article already categorized"); + return $article; + } - if (isset($ref["category_id"])) return $article; // already categorized + $nbs = new NaiveBayesianStorage($owner_uid); + $nb = new NaiveBayesianNgram($nbs, 3); $categories = $nbs->getCategories(); |