blallo/tt-rss
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

bayes: fix check for already categorized articles

Browse source
This commit is contained in:
Andrew Dolgov 2015-06-18 10:56:42 +03:00
parent e0940bfc4f
commit 5df13f31ef
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
plugins/af_sort_bayes/init.php
View file

@ -263,13 +263,18 @@ class Af_Sort_Bayes extends Plugin {
function hook_article_filter($article) {
$owner_uid = $article["owner_uid"];
// 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']) . "'");
if (db_num_rows($result) != 0) {
_debug("bayes: article already categorized");
return $article;
}
$nbs = new NaiveBayesianStorage($owner_uid);
$nb = new NaiveBayesianNgram($nbs, 3);
$ref = $nbs->getReference($article["guid"], false);
if (isset($ref["category_id"])) return $article; // already categorized
$categories = $nbs->getCategories();
if (count($categories) > 0) {