Lan Zagar avatar Lan Zagar committed ea84231

Corrected probabilities for binary class prediction.

Comments (0)

Files changed (1)

_multitask/mtfeat.py

         f = dot(self.W[:, t], ins.native()[:-1])
         if self.intercepts:
             f += self.intercepts[t]
+        dist = Orange.statistics.distribution.Distribution(self.domain.class_var)
         if isinstance(self.domain.class_var, Orange.feature.Continuous):
             val = self.domain.class_var(f)
+            dist[val] = 1.
         else:
             val = self.domain.class_var(int(f > 0))
-        dist = Orange.statistics.distribution.Distribution(val.variable)
-        dist[val] = 1. / (1. + np.exp(-f))
+            dist[1] = 1. / (1. + np.exp(-f))
+            dist[0] = 1 - dist[0]
         if return_type == Orange.core.GetValue:
             return val
         elif return_type == Orange.core.GetBoth:
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.