generated from LufsX/Experiment-LogReg
This commit is contained in:
+4
-21
@@ -44,18 +44,14 @@ def train_logreg(X, y, learning_rate=0.01, iterations=1000):
|
|||||||
weights = np.zeros(n)
|
weights = np.zeros(n)
|
||||||
|
|
||||||
for _ in range(iterations):
|
for _ in range(iterations):
|
||||||
# 计算预测值
|
|
||||||
z = np.dot(X, weights)
|
z = np.dot(X, weights)
|
||||||
predictions = 1 / (1 + np.exp(-z))
|
predictions = 1 / (1 + np.exp(-z))
|
||||||
|
|
||||||
# 计算损失的梯度
|
|
||||||
gradient = np.dot(X.T, (predictions - y)) / m
|
gradient = np.dot(X.T, (predictions - y)) / m
|
||||||
|
|
||||||
# 更新权重
|
|
||||||
weights -= learning_rate * gradient
|
weights -= learning_rate * gradient
|
||||||
|
|
||||||
return weights
|
return weights
|
||||||
|
|
||||||
|
|
||||||
def predict_logreg(inX, weights):
|
def predict_logreg(inX, weights):
|
||||||
"""
|
"""
|
||||||
对率回归预测函数。
|
对率回归预测函数。
|
||||||
@@ -67,9 +63,8 @@ def predict_logreg(inX, weights):
|
|||||||
Returns:
|
Returns:
|
||||||
- int: 预测的分类标签。
|
- int: 预测的分类标签。
|
||||||
"""
|
"""
|
||||||
z = np.dot(inX, weights)
|
z = np.dot(inX, weights)
|
||||||
prediction_prob = 1 / (1 + np.exp(-z))
|
prediction_prob = 1 / (1 + np.exp(-z))
|
||||||
|
|
||||||
return 1 if prediction_prob >= 0.5 else 0
|
return 1 if prediction_prob >= 0.5 else 0
|
||||||
|
|
||||||
# 这里也写自己的代码实现
|
# 这里也写自己的代码实现
|
||||||
@@ -82,25 +77,13 @@ def load_data(filepath):
|
|||||||
values = line.strip().replace('?', '0').split()
|
values = line.strip().replace('?', '0').split()
|
||||||
try:
|
try:
|
||||||
values = [float(v) for v in values]
|
values = [float(v) for v in values]
|
||||||
data.append(values[:-1])
|
data.append(values[:-1]) # 特征
|
||||||
labels.append(values[-1])
|
labels.append(values[-1]) # 标签
|
||||||
except ValueError:
|
except ValueError:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
return np.array(data), np.array(labels)
|
return np.array(data), np.array(labels)
|
||||||
|
|
||||||
def preprocess_data(filepath):
|
|
||||||
# 加载数据
|
|
||||||
X, y = load_data(filepath)
|
|
||||||
|
|
||||||
# 特征缩放
|
|
||||||
scaler = StandardScaler()
|
|
||||||
X_scaled = scaler.fit_transform(X)
|
|
||||||
|
|
||||||
# 拆分数据集
|
|
||||||
X_train, X_test, y_train, y_test = train_test_split(X_scaled, y, test_size=0.2, random_state=42)
|
|
||||||
|
|
||||||
return X_train, X_test, y_train, y_test
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
# 以下代码可用于测试,可自行随意修改
|
# 以下代码可用于测试,可自行随意修改
|
||||||
|
|||||||
Reference in New Issue
Block a user