Doctorado-ML/stree
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/STree.git synced 2025-08-16 16:06:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Mfstomufs #43 (#44)

Browse Source 
* Implement module mfs changed name to mufs

* Update github CI file
This commit is contained in:
Ricardo Montañana Gómez
2021-08-02 18:03:59 +02:00
committed by GitHub
parent fc9b7b5c92
commit 0afe14a447
7 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
stree/Splitter.py
View File

@@ -12,7 +12,7 @@ from sklearn.feature_selection import SelectKBest, mutual_info_classif
from sklearn.preprocessing import StandardScaler
from sklearn.svm import SVC
from sklearn.exceptions import ConvergenceWarning
from mfs import MFS
from mufs import MUFS
class Snode:
@@ -312,8 +312,8 @@ class Splitter:
tuple
indices of the features selected
"""
mfs = MFS(max_features=max_features, discrete=False)
return mfs.cfs(dataset, labels).get_results()
mufs = MUFS(max_features=max_features, discrete=False)
return mufs.cfs(dataset, labels).get_results()
@staticmethod
def _fs_fcbf(
@@ -336,8 +336,8 @@ class Splitter:
tuple
indices of the features selected
"""
mfs = MFS(max_features=max_features, discrete=False)
return mfs.fcbf(dataset, labels, 5e-4).get_results()
mufs = MUFS(max_features=max_features, discrete=False)
return mufs.fcbf(dataset, labels, 5e-4).get_results()
def partition_impurity(self, y: np.array) -> np.array:
return self.criterion_function(y)

2
stree/__init__.py
View File

@@ -1,6 +1,6 @@
from .Strees import Stree, Siterator
__version__ = "1.2"
__version__ = "1.2.1"
__author__ = "Ricardo Montañana Gómez"
__copyright__ = "Copyright 2020-2021, Ricardo Montañana Gómez"