multi hot 转化成数字

tensnsorflow/multi_hot.py

@@ -27,7 +27,7 @@ def multi():
     df["l2"] = "lost"
     df["l3"] = "lost"
     for i in list(df["level"].unique()):
-        l = i.split(";")
+        l = [int(j) for j in i.split(";")]
         l = sorted(l)
         if len(l) >= 3:
             df.loc[df["level"] == i, ["l1"]] = l[0]