def test_lsa_with_custom_indices(text_df):
X = text_df
X = X.set_index(pd.Series([2, 5, 19]))
lsa = LSA(text_columns=['col_1', 'col_2'])
lsa.fit(X)
X_t = lsa.transform(X)
assert not X_t.to_dataframe().isnull().any().any()
def test_lsa_only_text(text_df):
X = text_df
lsa = LSA(text_columns=['col_1', 'col_2'])
lsa.fit(X)
expected_col_names = set(
['LSA(col_1)[0]', 'LSA(col_1)[1]', 'LSA(col_2)[0]', 'LSA(col_2)[1]'])
X_t = lsa.transform(X)
assert set(X_t.columns) == expected_col_names
assert len(X_t.columns) == 4
assert X_t.dtypes.all() == np.float64
def test_some_missing_col_names(text_df, caplog):
X = text_df
expected_col_names = set(['LSA(col_1)[0]',
'LSA(col_1)[1]',
'LSA(col_2)[0]',
'LSA(col_2)[1]'])
lsa = LSA(text_columns=['col_1', 'col_2', 'col_3'])
lsa.fit(X)
X_t = lsa.transform(X)
assert set(X_t.columns) == expected_col_names
assert len(X_t.columns) == 4
assert set(X_t.logical_types.values()) == {ww.logical_types.Double}
def test_lsa_only_text(text_df):
X = text_df
lsa = LSA(text_columns=['col_1', 'col_2'])
lsa.fit(X)
expected_col_names = set(['LSA(col_1)[0]',
'LSA(col_1)[1]',
'LSA(col_2)[0]',
'LSA(col_2)[1]'])
X_t = lsa.transform(X)
assert set(X_t.columns) == expected_col_names
assert len(X_t.columns) == 4
assert set(X_t.logical_types.values()) == {ww.logical_types.Double}
def test_lsa_output():
X = pd.DataFrame(
{'lsa': ['do you hear the people sing? Singing the songs of angry men\n\tIt is the music of a people who will NOT be slaves again!',
'I dreamed a dream in days gone by, when hope was high and life worth living',
'Red, the blood of angry men - black, the dark of ages past']})
lsa = LSA(text_columns=['lsa'])
lsa.fit(X)
expected_features = pd.DataFrame([[0.832, 0.],
[0., 1.],
[0.832, 0.]], columns=["LSA(lsa)[0]", "LSA(lsa)[1]"])
X_t = lsa.transform(X)
cols = [col for col in X_t.columns if 'LSA' in col]
features = X_t[cols]
assert_frame_equal(expected_features, features.to_dataframe(), atol=1e-3)
def test_some_missing_col_names(text_df, caplog):
X = text_df
lsa = LSA(text_columns=['col_1', 'col_2', 'col_3'])
with caplog.at_level(logging.WARNING):
lsa.fit(X)
assert "Columns ['col_3'] were not found in the given DataFrame, ignoring" in caplog.messages
expected_col_names = set(
['LSA(col_1)[0]', 'LSA(col_1)[1]', 'LSA(col_2)[0]', 'LSA(col_2)[1]'])
X_t = lsa.transform(X)
assert set(X_t.columns) == expected_col_names
assert len(X_t.columns) == 4
assert X_t.dtypes.all() == np.float64
def test_lsa_with_nontext(text_df):
X = text_df
X['col_3'] = [73.7, 67.213, 92]
lsa = LSA(text_columns=['col_1', 'col_2'])
lsa.fit(X)
expected_col_names = set([
'LSA(col_1)[0]', 'LSA(col_1)[1]', 'LSA(col_2)[0]', 'LSA(col_2)[1]',
'col_3'
])
X_t = lsa.transform(X)
assert set(X_t.columns) == expected_col_names
assert len(X_t.columns) == 5
assert X_t.dtypes.all() == np.float64
def test_lsa_with_nontext(text_df):
X = text_df
X['col_3'] = [73.7, 67.213, 92]
lsa = LSA(text_columns=['col_1', 'col_2'])
lsa.fit(X)
expected_col_names = set(['LSA(col_1)[0]',
'LSA(col_1)[1]',
'LSA(col_2)[0]',
'LSA(col_2)[1]',
'col_3'])
X_t = lsa.transform(X)
assert set(X_t.columns) == expected_col_names
assert len(X_t.columns) == 5
assert set(X_t.logical_types.values()) == {ww.logical_types.Double}
def test_index_col_names():
X = np.array([['I\'m singing in the rain!$%^ do do do do do da do', 'do you hear the people sing?////////////////////////////////////'],
['just singing in the rain.................. \n', 'singing the songs of angry men\n'],
['\t\n\n\n\nWhat a glorious feelinggggggggggg, I\'m happy again!!! lalalalalalalalalalala', '\tIt is the music of a people who will NOT be slaves again!!!!!!!!!!!']])
lsa = LSA(text_columns=[0, 1])
lsa.fit(X)
expected_col_names = set(['LSA(0)[0]',
'LSA(0)[1]',
'LSA(1)[0]',
'LSA(1)[1]'])
X_t = lsa.transform(X)
assert set(X_t.columns) == expected_col_names
assert len(X_t.columns) == 4
assert set(X_t.logical_types.values()) == {ww.logical_types.Double}
def test_lsa_output():
X = pd.DataFrame({
'lsa': [
'do you hear the people sing? Singing the songs of angry men\n\tIt is the music of a people who will NOT be slaves again!',
'I dreamed a dream in days gone by, when hope was high and life worth living',
'Red, the blood of angry men - black, the dark of ages past'
]
})
lsa = LSA(text_columns=['lsa'])
lsa.fit(X)
expected_features = [[0.832, 0.], [0., 1.], [0.832, 0.]]
X_t = lsa.transform(X)
cols = [col for col in X_t.columns if 'LSA' in col]
features = X_t[cols]
np.testing.assert_almost_equal(features, expected_features, decimal=3)
def test_lsa_woodwork_custom_overrides_returned_by_components(X_df):
y = pd.Series([1, 2, 1])
override_types = [Integer, Double, Categorical, Boolean, NaturalLanguage]
X_df['text col'] = pd.Series(['this will be a natural language column because length', 'yay', 'hay'], dtype="string")
lsa = LSA()
for logical_type in override_types:
try:
X = ww.DataTable(X_df, logical_types={0: logical_type})
except TypeError:
continue
lsa.fit(X)
transformed = lsa.transform(X, y)
assert isinstance(transformed, ww.DataTable)
if logical_type == NaturalLanguage:
assert transformed.logical_types == {'LSA(0)[0]': Double, 'LSA(0)[1]': Double, 'LSA(text col)[0]': Double, 'LSA(text col)[1]': Double}
else:
assert transformed.logical_types == {0: logical_type, 'LSA(text col)[0]': Double, 'LSA(text col)[1]': Double}
def test_int_col_names():
X = pd.DataFrame(
{4.75: ['I\'m singing in the rain! Just singing in the rain, what a glorious feeling, I\'m happy again!',
'In sleep he sang to me, in dreams he came... That voice which calls to me, and speaks my name.',
'I\'m gonna be the main event, like no king was before! I\'m brushing up on looking down, I\'m working on my ROAR!'],
-1: ['do you hear the people sing? Singing the songs of angry men\n\tIt is the music of a people who will NOT be slaves again!',
'I dreamed a dream in days gone by, when hope was high and life worth living',
'Red, the blood of angry men - black, the dark of ages past']
})
lsa = LSA(text_columns=[4.75, -1])
lsa.fit(X)
expected_col_names = set(['LSA(4.75)[0]',
'LSA(4.75)[1]',
'LSA(-1)[0]',
'LSA(-1)[1]'])
X_t = lsa.transform(X)
assert set(X_t.columns) == expected_col_names
assert len(X_t.columns) == 4
assert set(X_t.logical_types.values()) == {ww.logical_types.Double}
def test_lsa_no_text():
X = pd.DataFrame({'col_1': [1, 2, 3], 'col_2': [4, 5, 6]})
lsa = LSA()
lsa.fit(X)
X_t = lsa.transform(X)
assert len(X_t.columns) == 2
