2

我正在寻找使用 imblearn 的 SMOTE 为机器学习算法生成合成样本。我有一些分类特征，我已使用 sklearn preprocessing.LabelEncoder 将其转换为整数。

我遇到的问题是，当我使用 smote 生成合成数据时，数据点变成浮点数，而不是分类数据所需的整数。

from collections import Counter
from sklearn.datasets import make_classification
from imblearn.over_sampling import SMOTE
import pandas as pd
from matplotlib import pyplot as plt
from sklearn.decomposition import PCA
import numpy as np
from sklearn import preprocessing

if __name__ == '__main__':
    df = pd.read_csv('resample.csv')
    y = df['Result']
    accounts = df['Account Number']
    df.drop('Result',axis=1,inplace=True)
    df.drop('Account Number', axis=1, inplace=True)

    df.fillna(value=0, inplace=True)

    le = preprocessing.LabelEncoder()
    le.fit(df['Distribution Partner'])
    print(le.classes_)
    df['Distribution Partner'] = le.transform(df['Distribution Partner'])
    print('Original dataset shape {}'.format(Counter(y)))
    sm = SMOTE(kind='regular')
    X_resampled, y_resampled = sm.fit_sample(df, y)
    np.savetxt('output.csv', X_resampled, delimiter=",")
    print('New dataset shape {}'.format(Counter(y_resampled)))

无论如何，我可以让 SMOTE 生成合成样本，但只能使用 0、1、2 等而不是 0.5、1.23、2.004 的值？

4

2 回答 2

2

很简单：使用 SMOTENC 代替 SMOTE。SMOTENC 可以处理分类和非分类特征。

示例代码：

from imblearn.over_sampling import SMOTENC`
obj = SMOTENC(categorical_features = [1,4])
ovsersampled_features, ovsersampled_target = obj.fit_sample(Features, Target)

[1,4] = 数据集分类列的索引。

*索引从 0 开始。

于 2019-05-14T12:29:54.700 回答
0

不幸的是，imblearn 的 SMOTE 实现仅适用于连续数据。在这里讨论。

于 2017-03-03T18:58:09.660 回答