generated from LufsX/Experiment-LogReg
This commit is contained in:
+13
-2
@@ -55,10 +55,21 @@ def predict_logreg(inX, weights):
|
|||||||
return 0
|
return 0
|
||||||
# 这里也写自己的代码实现
|
# 这里也写自己的代码实现
|
||||||
def load_data(filepath):
|
def load_data(filepath):
|
||||||
data = np.loadtxt(filepath, delimiter=' ')
|
data = np.loadtxt(filepath, delimiter=',')
|
||||||
|
# 将 '?' 替换为 np.nan,表示缺失值
|
||||||
|
data = np.where(data == '?', np.nan, data)
|
||||||
|
# 将非数值列转换为数值,这里假设非数值列是分类数据,可以使用 label encoding 或 one-hot encoding
|
||||||
|
# 这里我们使用 label encoding 作为示例
|
||||||
|
for i in range(data.shape[1]):
|
||||||
|
if data[:, i].dtype == np.object: # 如果是对象类型(字符串),则进行编码
|
||||||
|
data[:, i] = np.nan # 先将 '?' 替换为 np.nan
|
||||||
|
data[:, i] = pd.Categorical(data[:, i]).codes # 将分类数据转换为数值
|
||||||
|
# 将缺失值填充为 0 或者其他合适的值
|
||||||
|
data = np.nan_to_num(data, nan=0.0)
|
||||||
X = data[:, :-1]
|
X = data[:, :-1]
|
||||||
y = data[:, -1]
|
y = data[:, -1]
|
||||||
return X, y
|
return X, y.astype(int) # 确保标签是整数类型
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
# 以下代码可用于测试,可自行随意修改
|
# 以下代码可用于测试,可自行随意修改
|
||||||
|
|||||||
Reference in New Issue
Block a user