• 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 pytest

import sklearn.datasets as datasets
import sklearn.kernel_approximation as ka

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


class TestKernelApproximation(tm.TestCase):

    def test_objectmapper(self):
        df = pdml.ModelFrame([])
        self.assertIs(df.kernel_approximation.AdditiveChi2Sampler,
                      ka.AdditiveChi2Sampler)
        self.assertIs(df.kernel_approximation.Nystroem, ka.Nystroem)
        self.assertIs(df.kernel_approximation.RBFSampler, ka.RBFSampler)
        self.assertIs(df.kernel_approximation.SkewedChi2Sampler,
                      ka.SkewedChi2Sampler)

    @pytest.mark.parametrize("algo", ['AdditiveChi2Sampler'])
    def test_Classifications(self, algo):
        iris = datasets.load_iris()
        df = pdml.ModelFrame(iris)

        mod1 = getattr(df.kernel_approximation, algo)()
        mod2 = getattr(ka, algo)()

        df.fit(mod1)
        mod2.fit(iris.data)

        result = df.transform(mod1)
        expected = mod2.transform(iris.data)

        self.assertIsInstance(result, pdml.ModelFrame)
        self.assert_numpy_array_almost_equal(result.data.values, expected)

    @pytest.mark.parametrize("algo", ['Nystroem', 'RBFSampler',
                                      'SkewedChi2Sampler'])
    def test_Classifications_Random(self, algo):
        iris = datasets.load_iris()
        df = pdml.ModelFrame(iris)

        mod1 = getattr(df.kernel_approximation, algo)(random_state=self.random_state)
        mod2 = getattr(ka, algo)(random_state=self.random_state)

        df.fit(mod1)
        mod2.fit(iris.data)

        result = df.transform(mod1)
        expected = mod2.transform(iris.data)

        self.assertIsInstance(result, pdml.ModelFrame)
        self.assert_numpy_array_almost_equal(result.data.values, expected)