Gabriel commited on
Commit
3fbae5e
·
1 Parent(s): caffb3a
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ def calc_preds(coeffs, indeps):
22
  if i != n-1:
23
  res = F.relu(res)
24
  # if torch.sigmoid(res) > 0.5:
25
- return res
26
 
27
  def main(job_title, company_name, company_desc, job_desc,
28
  job_requirement, salary, location, employment_type,
 
22
  if i != n-1:
23
  res = F.relu(res)
24
  # if torch.sigmoid(res) > 0.5:
25
+ return torch.sigmoid(res)
26
 
27
  def main(job_title, company_name, company_desc, job_desc,
28
  job_requirement, salary, location, employment_type,