Model baseline model imports to snorkel/labeling/model

vincentschen · vincentschen · commit 33b06da7cc38 · 2020-01-19T19:24:49.000Z
diff --git a/docs/packages/labeling.rst b/docs/packages/labeling.rst
@@ -14,12 +14,9 @@ Programmatic data set labeling: LF creation, models, and analysis utilities.
    LFApplier
    LabelModel
    LabelingFunction
-   MajorityClassVoter
-   MajorityLabelVoter
    lf.nlp.NLPLabelingFunction
    PandasLFApplier
    apply.dask.PandasParallelLFApplier
-   RandomVoter
    apply.spark.SparkLFApplier
    lf.nlp_spark.SparkNLPLabelingFunction
    filter_unlabeled_dataframe
diff --git a/snorkel/labeling/__init__.py b/snorkel/labeling/__init__.py
@@ -4,10 +4,5 @@
 from .apply.core import LFApplier  # noqa: F401
 from .apply.pandas import PandasLFApplier  # noqa: F401
 from .lf.core import LabelingFunction, labeling_function  # noqa: F401
-from .model.baselines import (  # noqa: F401
-    MajorityClassVoter,
-    MajorityLabelVoter,
-    RandomVoter,
-)
 from .model.label_model import LabelModel  # noqa: F401
 from .utils import filter_unlabeled_dataframe  # noqa: F401
diff --git a/snorkel/labeling/model/__init__.py b/snorkel/labeling/model/__init__.py
@@ -0,0 +1 @@
+from .baselines import MajorityClassVoter, MajorityLabelVoter, RandomVoter  # noqa: F401
diff --git a/test/labeling/model/test_baseline.py b/test/labeling/model/test_baseline.py
@@ -2,7 +2,7 @@
 
 import numpy as np
 
-from snorkel.labeling import MajorityClassVoter, MajorityLabelVoter, RandomVoter
+from snorkel.labeling.model import MajorityClassVoter, MajorityLabelVoter, RandomVoter
 
 
 class BaselineModelTest(unittest.TestCase):

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+from .baselines import MajorityClassVoter, MajorityLabelVoter, RandomVoter # noqa: F401`