handle the edge case of zero positives... sigh

2021-11-30 23:34:47 -08:00
parent fe96775e6e
commit 31e02f267c
1 changed files with 6 additions and 0 deletions
@@ -227,6 +227,12 @@ if __name__ == "__main__":
            print("skipping user %s, not %s" % (user, args.user))
            continue

+        # verify that we have at least one positive example...
+        num_papers_tagged = len(set().union(*tags.values()))
+        if num_papers_tagged == 0:
+            print("skipping user %s, no papers tagged" % (user, ))
+            continue
+
        # calculate the recommendations
        pids, scores = calculate_recommendation(tags, time_delta=args.time_delta)
        print("user %s has %d recommendations over last %d days" % (user, len(pids), args.time_delta))