Python DFtoVW.from_column_names Examples

Example #1

def test_from_column_names_no_label_constructor():
    df = pd.DataFrame({"x": [2]})
    conv = DFtoVW.from_column_names(x=["x"], df=df)
    lines_list = conv.convert_df()
    first_line = lines_list[0]
    assert first_line == "| x:2"

Example #2

def test_from_column_names_constructor():
    df = pd.DataFrame({"y": [1], "x": [2]})
    conv = DFtoVW.from_column_names(y="y", x=["x"], df=df)
    lines_list = conv.convert_df()
    first_line = lines_list[0]
    assert first_line == "1 | x:2"