Commit 7ea725f4 authored by schnurr's avatar schnurr
Browse files

Update probing.py

parent 2ec89e50

probing.py

+1 −1
Original line number Diff line number Diff line
@@ -115,7 +115,7 @@ def mask_first(pos_tag): 


        if pos_tag in pos_tags_sent:

            e = pos_tags_sent.index(pos_tag)

            df.iloc[i, 1] = df.iloc[i, 1].replace(pos_tag_tokens[e], random.randrange(1, 9))

            df.iloc[i, 1] = df.iloc[i, 1].replace(pos_tag_tokens[e], get_random_string(random.randrange(1, 9)))

        else:

            if i not in del_rows:

                del_rows.append(i)