Changeset 129:a87db93d93f6 for neighbors/neartag.py

Timestamp:

06/21/07 09:00:38 (14 months ago)

Author:

Tarek Ziad?? <tarek@…>

Message:

fixed tag calculation

Files:

• neighbors/neartag.py (modified) (2 diffs)

neighbors/neartag.py

@@ -4 +4 @@
 
     def __init__(self, tags):
+        self.tags = tags
         self.db = nearest.DataSet(attrnames=tags, exclude=(-1,),)
         self.solver = nearest.NearestNeighborLearner()
@@ -9 +10 @@
 
     def add_user(self, name, tags):
-        self.db.add_example(tags+[name])
+        values = []
+        for tag in self.tags:
+            if tag in tags:
+                values.append('1')
+            else:
+                values.append('0')
+        values.append(name)
+
+        self.db.add_example(values)
 
     def neighbours(self, name, k=10):