OpenMined / SyferText

A privacy preserving NLP framework
Apache License 2.0
198 stars 49 forks source link

Support for adding a single label classifier to a pipeline #168

Closed AlanAboudib closed 4 years ago

AlanAboudib commented 4 years ago

Description

This PR implements the SingleLabelClassifier class which uses an AverageDocEncoder.

It also supports adding the SingleLabelClassifier to the pipeline which could then be deployed to the Grid and used for inference on strings situated on any worker.

In this PR, the classifier is downloaded to the worker were data is situated. In a future PR, remote inference where the classifier is not downloaded to the data owner's worker is to be supported.

Fixes: #134 #135 #136 #137

Affected Dependencies

List any dependencies that are required for this change.

How has this been tested?

Checklist