函数内部的Scikit学习方法

时间:2018-09-07 10:18:58

标签: python scikit-learn

我具有以下使用sklearn.preprocessing特定方法的功能。

from sklearn import preprocessing
def process (data):
    method = preprocessing.MinMaxScaler()
    tranf = method.fit_transform(data)
    return tranf

我想拥有一个能够从sklearn.preprocessing调用任何方法的函数。看起来会这样。

from sklearn import preprocessing
def process (data, whichmethod):
    method = preprocessing.whichmethod()
    tranf = method.fit_transform(data)
    return tranf

请教我关于python和道歉的知识,因为我确实觉得这非常简单。

1 个答案:

答案 0 :(得分:2)

我认为现在传递正确的对象要比preprocessing中的名称容易得多:

from sklearn import preprocessing
def process (data, method):
    tranf = method.fit_transform(data)
    return tranf
process(data, preprocessing.MinMaxScaler()) # how you would reproduce your first example
相关问题