Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion few/evaluation.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,9 @@
import sys
from sklearn.metrics.pairwise import pairwise_distances
# from profilehooks import profile
from sklearn.externals.joblib import Parallel, delayed
from joblib import Parallel, delayed
from sklearn.feature_selection import f_classif, f_regression

# safe division
def divs(x,y):
"""safe division"""
Expand Down
7 changes: 4 additions & 3 deletions few/few.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,8 @@
from sklearn.pipeline import Pipeline
from sklearn.model_selection import cross_val_score, train_test_split, KFold
from sklearn.metrics import r2_score, accuracy_score, roc_auc_score
from sklearn.preprocessing import Imputer, StandardScaler
from sklearn.preprocessing import StandardScaler
from sklearn.impute import SimpleImputer
from sklearn.utils import check_random_state
from DistanceClassifier import DistanceClassifier
import numpy as np
Expand All @@ -34,7 +35,7 @@
import pdb
from collections import defaultdict
# from update_checker import update_check
from sklearn.externals.joblib import Parallel, delayed
from joblib import Parallel, delayed
from tqdm import tqdm
import uuid

Expand Down Expand Up @@ -433,7 +434,7 @@ def transform(self,x,inds=None,labels = None):

def impute_data(self,x):
"""Imputes data set containing Nan values"""
imp = Imputer(missing_values='NaN', strategy='mean', axis=0)
imp = SimpleImputer(missing_values='NaN', strategy='mean', axis=0)
return imp.fit_transform(x)

def clean(self,x):
Expand Down
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,10 +65,10 @@ def calculate_version():
This project is hosted at https://github.com/lacava/few
''',
zip_safe=True,
install_requires=['numpy', 'scipy', 'pandas', 'scikit-learn',
install_requires=['numpy', 'scipy', 'pandas', 'scikit-learn', 'joblib',
'update_checker', 'tqdm', 'joblib','DistanceClassifier',
'scikit-mdr','Cython', 'eigency'],
setup_requires=['numpy', 'scipy', 'pandas', 'scikit-learn',
setup_requires=['numpy', 'scipy', 'pandas', 'scikit-learn', 'joblib',
'update_checker', 'tqdm', 'joblib','DistanceClassifier',
'scikit-mdr','Cython', 'eigency'],
classifiers=[
Expand Down