• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• pandas-ml-master
  • requirements_test.txt
  • examples
    • kaggle_otto_xgboost.ipynb
    • kaggle_titanic.ipynb
  • LICENSE
  • pandas_ml
    • imbaccessors
      • test
        • test_imbalance.py
        • __init__.py
      • __init__.py
      • base.py
    • version.py
    • test
      • test_frame.py
      • test_groupby.py
      • __init__.py
      • test_series.py
    • snsaccessors
      • test
        • __init__.py
        • test_sns.py
      • __init__.py
      • base.py
    • util
      • testing.py
      • _util.py
      • __init__.py
    • misc
      • patsy_wraps.py
      • test
        • __init__.py
        • test_patsy.py
      • __init__.py
    • tools.py
    • xgboost
      • test
        • test_base.py
        • __init__.py
      • __init__.py
      • base.py
    • smaccessors
      • test
        • test_base.py
        • __init__.py
        • test_datasets.py
      • __init__.py
      • base.py
    • __init__.py
    • core
      • groupby.py
      • accessor.py
      • frame.py
      • generic.py
      • series.py
      • __init__.py
      • base.py
    • confusion_matrix
      • cm.py
      • test
        • test_confusion_matrix.py
        • __init__.py
      • stats.py
      • __init__.py
      • abstract.py
      • bcm.py
    • compat.py
    • skaccessors
      • pipeline.py
      • gaussian_process.py
      • ensemble.py
      • metrics.py
      • test
        • test_decomposition.py
        • test_calibration.py
        • test_base.py
        • test_neighbors.py
        • test_cluster.py
        • test_feature_selection.py
        • test_covariance.py
        • test_discriminant_analysis.py
        • test_random_projection.py
        • test_gaussian_process.py
        • test_preprocessing.py
        • test_ensemble.py
        • test_manifold.py
        • test_multiclass.py
        • test_svm.py
        • test_linear_model.py
        • test_isotonic.py
        • test_naive_bayes.py
        • test_multioutput.py
        • test_dummy.py
        • test_cross_decomposition.py
        • test_mixture.py
        • test_feature_extraction.py
        • test_kernel_approximation.py
        • __init__.py
        • test_metrics.py
        • test_pipeline.py
        • test_datasets.py
        • test_tree.py
        • test_kernel_ridge.py
        • test_semi_supervised.py
        • test_model_selection.py
        • test_neural_network.py
      • linear_model.py
      • preprocessing.py
      • feature_extraction.py
      • model_selection.py
      • decomposition.py
      • manifold.py
      • __init__.py
      • cluster.py
      • svm.py
      • feature_selection.py
      • covariance.py
      • neighbors.py
      • cross_decomposition.py
      • isotonic.py
      • base.py
  • README.rst
  • setup.py
  • .travis.yml
  • scripts
    • build_travis.sh
  • requirements.txt
  • LICENSES
    • SEABORN_LICENSE
    • SKLEARN_LICENSE
    • PANDAS_LICENSE
    • XGBOOST_LICENSE
  • .gitignore
  • MANIFEST.in
  • doc
    • Makefile
    • source
      • whatsnew.rst
      • patsy.rst
      • modelframe.rst
      • sklearn.rst
      • imbalance.rst
      • screenshots
      • index.rst
      • conf.py
      • conf_mat.rst
      • xgboost.rst
    • make.bat

#!/usr/bin/env python

import sklearn.feature_extraction as fe

import pandas_ml as pdml
import pandas_ml.util.testing as tm


class TestFeatureExtraction(tm.TestCase):

    def test_objectmapper(self):
        df = pdml.ModelFrame([])
        self.assertIs(df.feature_extraction.DictVectorizer, fe.DictVectorizer)
        self.assertIs(df.feature_extraction.FeatureHasher, fe.FeatureHasher)

        self.assertIs(df.feature_extraction.image.img_to_graph, fe.image.img_to_graph)
        self.assertIs(df.feature_extraction.image.grid_to_graph, fe.image.grid_to_graph)
        self.assertIs(df.feature_extraction.image.extract_patches_2d, fe.image.extract_patches_2d)
        self.assertIs(df.feature_extraction.image.reconstruct_from_patches_2d,
                      fe.image.reconstruct_from_patches_2d)
        self.assertIs(df.feature_extraction.image.PatchExtractor, fe.image.PatchExtractor)

        self.assertIs(df.feature_extraction.text.CountVectorizer, fe.text.CountVectorizer)
        self.assertIs(df.feature_extraction.text.HashingVectorizer, fe.text.HashingVectorizer)
        self.assertIs(df.feature_extraction.text.TfidfTransformer, fe.text.TfidfTransformer)
        self.assertIs(df.feature_extraction.text.TfidfVectorizer, fe.text.TfidfVectorizer)