generated from LufsX/Experiment-LogReg
This commit is contained in:
+3
-3
@@ -87,15 +87,15 @@ def load_data(filepath):
|
|||||||
# 检查特征数量
|
# 检查特征数量
|
||||||
num_features = data.shape[1]
|
num_features = data.shape[1]
|
||||||
|
|
||||||
# 特征选择:去掉特征3(索引2)和特征24~28(索引23~26),确保索引有效
|
# 特征选择:去掉特征3和特征24~28,确保索引有效
|
||||||
indices_to_delete = [2] # 特征3
|
indices_to_delete = [2] # 特征3
|
||||||
if num_features > 24: # 确保特征数量足够
|
if num_features > 24: # 确保特征数量足够
|
||||||
indices_to_delete += list(range(24, 27)) # 特征24到27(索引23到26)
|
indices_to_delete += list(range(24, 27)) # 特征24到27
|
||||||
|
|
||||||
# 删除特征
|
# 删除特征
|
||||||
data = np.delete(data, indices_to_delete, axis=1)
|
data = np.delete(data, indices_to_delete, axis=1)
|
||||||
|
|
||||||
# 将标签二分类化,特征23(索引22)取值2和3合并为0,其余为1
|
# 将标签二分类化,特征23取值2和3合并为0,其余为1
|
||||||
target_variable = np.where((labels == 2) | (labels == 3), 0, 1)
|
target_variable = np.where((labels == 2) | (labels == 3), 0, 1)
|
||||||
|
|
||||||
return data, target_variable
|
return data, target_variable
|
||||||
|
|||||||
Reference in New Issue
Block a user