我是 Python 新手，如果太簡單，請提前致歉。我的代碼是# Split datay = starbucks_smote.iloc[:, -1]X = starbucks_smote.drop('label', axis = 1)# Count labels by typecounter = Counter(y)print(counter)Counter({0: 9634, 1: 2895})# Transform the datasetoversample = SMOTE()X, y = oversample.fit_resample(X, y)# Print the oversampled datasetcounter = Counter(y)print(counter)Counter({0: 9634, 1: 9634})如何保存過采樣數據集以備將來使用？我試過data_res = np.concatenate((X, y), axis = 1)data_res.to_csv('sample_smote.csv')出錯了ValueError: all the input arrays must have same number of dimensions,?but the array at index 0 has 2 dimension(s) and the array at index 1 has 1 dimension(s)感謝任何提示！
1 回答
紫衣仙女
TA貢獻1839條經驗 獲得超15個贊
您可以創建數據框：
data_res?=?pd.DataFrame(X) data_res['y']?=?y
然后保存data_res到 CSV。
基于連接 od 的解決方案numpy.arrays也是可能的，但np.vstack需要使維度兼容：
data_res?=?np.concatenate((X,?np.vstack(y)),?axis?=?1) data_res?=?pd.DataFrame(data_res)
添加回答
舉報
0/150
提交
取消