Changeset 159:e4e3160be9c0 for classifier/classifier.py

Timestamp:

08/19/07 10:34:17 (15 months ago)

Author:

Tarek Ziad?? <tarek@…>

Message:

refactoring in progress

Files:

• classifier/classifier.py (modified) (3 diffs)

classifier/classifier.py

@@ -42 +42 @@
         wich means: store words in categories
         """
-        self.backend.addCategory(name=category, title=category)
+        self.backend.add_category(name=category)
         data = self.tokenizer.transform(data, self.options)
         for element in data:
-            self.backend.addWord(element, self.language, category)
+            self.backend.add_word(element, self.language, category)
 
     def unlearn(self, data, category):
@@ -116 +116 @@
 
     def corpusSize(self, language=None):
-        return self.backend.wordCount(language=language)
+
+        return self.backend.word_count(language=language)
 
     def categorySize(self, category, language=None):
-        return self.backend.wordCount(category=category, language=language)
+        return self.backend.word_count(category=category, language=language)
 
     def _buildWordProbabilities(self, language=None):
         probs = {}
-        for cat in self.backend.listCategories():
+        for cat in self.backend.list_categories():
             probs[cat] = self._buildCategoryWordProbabilities(cat, language)
         return probs
@@ -138 +139 @@
         them_count = float(max(corpus_size - category_size, 1))
         probabilities = {}
-        words = self.backend.listWords(language, complete=True)
+        words = self.backend.list_words(language, complete=True)
 
         for word in words: