Sklearn wrapper integration

gensim/sklearn_integration/__init__.py

@@ -0,0 +1,6 @@
+"""scikit learn wrapper for gensim
+Contains various gensim based implementations
+which match with scikit-learn standards .
+See [1] for complete set of conventions.
+[1] http://scikit-learn.org/stable/developers/
+"""

gensim/sklearn_integration/base.py

@@ -0,0 +1,9 @@
+"""
+scikit learn interface for gensim for easy use of gensim with scikit-learn
+"""
+import numpy as np
+class BaseClass(object):
+    def __init__(self):
+        """init"""
+    def run(self):
+        return np.array([0,0,0])

gensim/test/test_sklearn_integration.py

@@ -0,0 +1,32 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2010 Radim Rehurek <[email protected]>
+# Licensed under the GNU LGPL v2.1 - http://www.gnu.org/licenses/lgpl.html
+
+"""
+Tests for sklearn_integration interface
+"""
+
+
+import logging
+import unittest
+import os
+import os.path
+import numpy as np
+
+from gensim.sklearn_integration import base
+
+
+class TestSklearn(unittest.TestCase):
+    """
+    write test script
+    """
+    #for now
+    def testRun(self):
+        model=base.BaseClass()
+        self.assertTrue(np.array_equal(model.run(),np.array([0,0,0])))
+
+if __name__ == '__main__':
+    logging.basicConfig(format='%(asctime)s : %(levelname)s : %(message)s', level=logging.DEBUG)
+    unittest.main()