Commits

Lan Zagar  committed ea84231

Corrected probabilities for binary class prediction.

  • Participants
  • Parent commits 7964d8d

Comments (0)

Files changed (1)

File _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: