Commit b98b133d authored by schnurr's avatar schnurr
Browse files

formatting

parent 7fa82293

src/probing.py

+3 −1
Original line number Diff line number Diff line
@@ -69,7 +69,8 @@ def get_random_string(length): 


def mask_token(pos_tag):

    """

    For every sentence in the dataframe, mask that token whose POS matches the formal parameter and is also the highest in the dependency tree's structure. If there's more than one option, take that token which comes first in the sentence.

    For every sentence in the dataframe, mask that token whose POS matches the formal parameter and is also the highest in the dependency tree's structure. 

    If there's more than one option, take that token which comes first in the sentence.

    Return the changed dataframe.

    """

    df = STSB_Dataset('../data/stsbenchmark/sts-test.csv').as_dataframe
@@ -174,3 +175,4 @@ def mask_root(): 
        

    return df