From e0940bfc4fdfad943a111b52569a1917e083b2c0 Mon Sep 17 00:00:00 2001 From: Andrew Dolgov Date: Thu, 18 Jun 2015 09:42:20 +0300 Subject: [PATCH] experiment: switch to 3-gram tokens --- plugins/af_sort_bayes/init.php | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/plugins/af_sort_bayes/init.php b/plugins/af_sort_bayes/init.php index fdb573ecd..d2e659cae 100644 --- a/plugins/af_sort_bayes/init.php +++ b/plugins/af_sort_bayes/init.php @@ -39,7 +39,7 @@ class Af_Sort_Bayes extends Plugin { $dst_category = "UGLY"; $nbs = new NaiveBayesianStorage($_SESSION["uid"]); - $nb = new NaiveBayesian($nbs); + $nb = new NaiveBayesianNgram($nbs, 3); $result = $this->dbh->query("SELECT score, guid, title, content FROM ttrss_entries, ttrss_user_entries WHERE ref_id = id AND id = " . $article_id . " AND owner_uid = " . $_SESSION["uid"]); @@ -264,7 +264,7 @@ class Af_Sort_Bayes extends Plugin { $owner_uid = $article["owner_uid"]; $nbs = new NaiveBayesianStorage($owner_uid); - $nb = new NaiveBayesian($nbs); + $nb = new NaiveBayesianNgram($nbs, 3); $ref = $nbs->getReference($article["guid"], false); @@ -336,7 +336,7 @@ class Af_Sort_Bayes extends Plugin { $this->dbh->query("COMMIT"); $nbs = new NaiveBayesianStorage($_SESSION["uid"]); - $nb = new NaiveBayesian($nbs); + $nb = new NaiveBayesianNgram($nbs, 3); $nb->updateProbabilities(); }