Confidence Interval

INGEOTEC · May 25, 2021 · 0576947 · 0576947
1 parent 973fda6
commit 0576947
Show file tree

Hide file tree

Showing 7 changed files with 103 additions and 349 deletions.
diff --git a/EvoMSA/__init__.py b/EvoMSA/__init__.py
@@ -11,4 +11,4 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-__version__ = '1.3.1'
+__version__ = '1.3.2'
diff --git a/EvoMSA/model_selection.py b/EvoMSA/model_selection.py
diff --git a/EvoMSA/tests/test_model_selection.py b/EvoMSA/tests/test_model_selection.py
diff --git a/EvoMSA/tests/test_utils.py b/EvoMSA/tests/test_utils.py
@@ -44,4 +44,14 @@ def test_cache_cl():
     for i, j in zip(cache.ml_kfold(), ll):
         print(i)
         assert i == "hola-%s-%s-K" % (j, "LinearSVC")
-
+
+
+def test_confidence_interval():
+    from EvoMSA.utils import ConfidenceInterval
+    from EvoMSA.tests.test_base import get_data
+
+    X, y = get_data()
+    kw = dict(stacked_method="sklearn.naive_bayes.GaussianNB") 
+    ci = ConfidenceInterval(X, y, evomsa_kwargs=kw)
+    result = ci.estimate()
+    assert len(result) == 2