def testSingleCategoricalProcessorPandasSingleDF(self):
if HAS_PANDAS:
import pandas as pd
cat_processor = categorical.CategoricalProcessor()
data = pd.DataFrame({"Gender": ["Male", "Female", "Male"]})
x = list(cat_processor.fit_transform(data))
self.assertAllEqual(list(x), [[1], [2], [1]])
def testSingleCategoricalProcessor(self):
cat_processor = categorical.CategoricalProcessor(
min_frequency=1)
X = cat_processor.fit_transform(
[["0"], [1], [float('nan')],
["C"], ["C"], [1], ["0"], [np.nan], [3]])
self.assertAllEqual(list(X), [
[2], [1], [0], [3],
[3], [1], [2], [0],
[0]])
def testMultiCategoricalProcessor(self):
cat_processor = categorical.CategoricalProcessor(min_frequency=0,
share=False)
x = cat_processor.fit_transform([["0", "Male"], [1, "Female"], ["3", "Male"]
])
self.assertAllEqual(list(x), [[1, 1], [2, 2], [3, 1]])