precompute_probs.py 文件源码

def main():
  parser = argparse.ArgumentParser()
  parser.add_argument('--tag', default='pairs')
  parser.add_argument('--fold', default='test')
  args = parser.parse_args()

  #metavec = load_metavectors(args.fold)

  clf = train.load_model(args.tag)
  X, y = vectorize.load_fold(args.fold, args.tag)

  if hasattr(clf, 'predict_proba'):
    probs = clf.predict_proba(X)
    # returns an array of shape (n, 2), where each len-2 subarray
    # has the probability of the negative and positive classes. which is silly.
    probs = probs[:,1]
  else:
    scores = clf.decision_function(X)
    probs = expit(scores)

  pdict = pdictify(probs, args.fold)
  common.save_pdict_for_tag(args.tag, pdict, args.fold)