|
1 | | -from sklearn.datasets import load_iris |
2 | | -import os |
3 | | -from zipfile import ZipFile |
4 | | -from fuse.utils.file_io.file_io import create_dir, read_dataframe |
5 | | -import wget |
6 | | -from typing import Hashable, Optional, Sequence, List, Tuple |
7 | | -import torch |
8 | | -from scipy.io import arff |
| 1 | +from typing import List |
9 | 2 | import pandas as pd |
10 | | -import numpy as np |
11 | 3 |
|
12 | | -from fuse.data import DatasetDefault |
13 | | -from fuse.data.ops.ops_cast import OpToTensor, OpToNumpy, OpToInt |
14 | | -from fuse.data.utils.sample import get_sample_id |
15 | | -from fuse.data.pipelines.pipeline_default import PipelineDefault |
16 | | -from fuse.data.ops.op_base import OpBase |
17 | | -from fuse.data.datasets.caching.samples_cacher import SamplesCacher |
18 | | -from fuse.data.ops.ops_aug_common import OpSample |
19 | | -from fuse.data.ops.ops_read import OpReadDataframe |
20 | | -from fuse.data.ops.ops_common import OpLambda, OpOverrideNaN |
21 | | -from fuseimg.data.ops.color import OpToRange, OpNormalizeAgainstSelf |
22 | | -from fuse.data.ops.ops_debug import OpPrintKeys, OpPrintKeysContent, OpPrintShapes, OpPrintTypes |
23 | | -from fuseimg.data.ops.ops_debug import OpVis2DImage |
24 | | - |
25 | | -from fuse.utils import NDict |
26 | | - |
27 | | -from fuseimg.data.ops.image_loader import OpLoadImage |
28 | | -from fuseimg.data.ops.aug.color import OpAugColor, OpAugGaussian |
29 | | -from fuseimg.data.ops.aug.geometry import OpResizeTo, OpAugAffine2D |
30 | | -from fuse.utils.rand.param_sampler import Uniform, RandInt, RandBool |
31 | 4 | from sklearn.feature_selection import SelectKBest, mutual_info_classif |
32 | 5 |
|
33 | | -from ops.ops_sagi import * |
34 | | -from ops.ops_shaked import * |
35 | | -import skimage |
36 | | - |
37 | 6 |
|
38 | 7 | def feature_selection(data: pd.DataFrame, k: int) -> List[str]: |
39 | 8 | X = data.drop("0", axis=1) |
|
0 commit comments