allennlp-dataframe-mapper 0.0.3

No project description provided

sklearn-pandas plugin for AllenNLP

sklearn-pandas plugin / wrapper for AllenNLP.

Install

$ pip install git+ssh://[e-mail]/shunk031/allennlp-dataframe-mapper.git

Usage

Config

mapper_iris.jsonnet

{
    "type": "default",
    "features": [
        [["sepal length (cm)"], null],
        [["sepal width (cm)"], null],
        [["petal length (cm)"], null],
        [["petal width (cm)"], null],
        [["species"], [{"type": "flatten"}, {"type": "label-encoder"}]],
    ],
    "df_out": true,
}

Mapper

from allennlp.common import Params
from allennlp_dataframe_mapper import DataFrameMapper

params = Params.from_file("mapper_iris.jsonnet")
mapper = DataFrameMapper.from_params(params=params)

print(mapper)
# DataFrameMapper(df_out=True,
#                 features=[(['sepal length (cm)'], None, {}),
#                           (['sepal width (cm)'], None, {}),
#                           (['petal length (cm)'], None, {}),
#                           (['petal width (cm)'], None, {}),
#                           (['species'], [FlattenTransform(), LabelEncoder()], {})])

mapper.fit_transform(df)

License

MIT