我试图在Python中计算缺失值,而sklearn
似乎没有超出平均值(均值,中位数或模式)估算的方法。 Orange imputation model似乎提供了一个可行的选择。但是,Orange.data.Table
似乎无法识别np.nan
或某种方式导致归因失败。
import Orange
import numpy as np
tmp = np.array([[1, 2, np.nan, 5, 8, np.nan], [40, 4, 8, 1, 0.2, 9]])
data = Orange.data.Table(tmp)
imputer = Orange.feature.imputation.ModelConstructor()
imputer.learner_continuous = Orange.classification.tree.TreeLearner(min_subset=20)
imputer = imputer(data )
impdata = imputer(data)
for i in range(0, len(tmp)):
print impdata[i]
输出
[1.000, 2.000, 1.#QO, 5.000, 8.000, 1.#QO]
[40.000, 4.000, 8.000, 1.000, 0.200, 9.000]
知道我错过了什么吗?谢谢!
答案 0 :(得分:3)
似乎问题是Orange中的缺失值表示为?
或~
。奇怪的是,Orange.data.Table(numpy.ndarray)
构造函数并未推断numpy.nan
应转换为?
或~
,而是将其转换为1.#QO
。下面的自定义函数pandas_to_orange()
解决了这个问题。
import Orange
import numpy as np
import pandas as pd
from collections import OrderedDict
# Adapted from https://github.com/biolab/orange3/issues/68
def construct_domain(df):
columns = OrderedDict(df.dtypes)
def create_variable(col):
if col[1].__str__().startswith('float'):
return Orange.feature.Continuous(col[0])
if col[1].__str__().startswith('int') and len(df[col[0]].unique()) > 50:
return Orange.feature.Continuous(col[0])
if col[1].__str__().startswith('date'):
df[col[0]] = df[col[0]].values.astype(np.str)
if col[1].__str__() == 'object':
df[col[0]] = df[col[0]].astype(type(""))
return Orange.feature.Discrete(col[0], values = df[col[0]].unique().tolist())
return Orange.data.Domain(list(map(create_variable, columns.items())))
def pandas_to_orange(df):
domain = construct_domain(df)
df[pd.isnull(df)]='?'
return Orange.data.Table(Orange.data.Domain(domain), df.values.tolist())
df = pd.DataFrame({'col1':[1, 2, np.nan, 4, 5, 6, 7, 8, 9, np.nan, 11],
'col2': [10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110.]})
tmp = pandas_to_orange(df)
for i in range(0, len(tmp)):
print tmp[i]
输出结果为:
[1.000, 10.000]
[2.000, 20.000]
[?, 30.000]
[4.000, 40.000]
[5.000, 50.000]
[6.000, 60.000]
[7.000, 70.000]
[8.000, 80.000]
[9.000, 90.000]
[?, 100.000]
[11.000, 110.000]
我想正确编码缺失值的原因是我可以使用Orange插补库。但是,看起来库中的预测树模型并不比简单的平均插补更有效。具体来说,它会为所有缺失值计算相同的值。
imputer = Orange.feature.imputation.ModelConstructor()
imputer.learner_continuous = Orange.classification.tree.TreeLearner(min_subset=20)
imputer = imputer(tmp )
impdata = imputer(tmp)
for i in range(0, len(tmp)):
print impdata[i]
这是输出:
[1.000, 10.000]
[2.000, 20.000]
[5.889, 30.000]
[4.000, 40.000]
[5.000, 50.000]
[6.000, 60.000]
[7.000, 70.000]
[8.000, 80.000]
[9.000, 90.000]
[5.889, 100.000]
[11.000, 110.000]
我正在寻找适合模型的东西,比如kNN,对整个案例,并使用拟合模型来预测遗失的案例。 fancyimpute
(a Python 3 package)执行此操作但会在我的300K +输入上抛出MemoryError
。
from fancyimpute import KNN
df = pd.DataFrame({'col1':[1, 2, np.nan, 4, 5, 6, 7, 8, 9, np.nan, 11],
'col2': [10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110.]})
X_filled_knn = KNN(k=3).complete(df)
X_filled_knn
输出是:
array([[ 1. , 10. ],
[ 2. , 20. ],
[ 2.77777784, 30. ],
[ 4. , 40. ],
[ 5. , 50. ],
[ 6. , 60. ],
[ 7. , 70. ],
[ 8. , 80. ],
[ 9. , 90. ],
[ 9.77777798, 100. ],
[ 11. , 110. ]])
我可能会找到一种解决方法或将数据集拆分成块(不理想)。
答案 1 :(得分:1)
在Orange v2中,您可以将numpy蒙面数组传递给Orange.data.Table构造函数。修改你的例子:
id