Commit 69f81891 authored by schnurr's avatar schnurr
Browse files

Update probing.py

parent f9ef0875

probing.py

+4 −0
Original line number Diff line number Diff line
@@ -63,6 +63,8 @@ def get_random_string(length): 


def mask_token(pos_tag):



    df = STSB_Dataset('data/stsbenchmark/sts-test.csv').as_dataframe

    

    for e in range(len(dependency_hierarchy1)):

        mask_token = None

        for layer in dependency_hierarchy1[e]:
@@ -90,6 +92,8 @@ def mask_token(pos_tag): 


def mask_first(pos_tag):



    df = STSB_Dataset('data/stsbenchmark/sts-test.csv').as_dataframe



    i = -1

    del_rows = []