generated from LufsX/Experiment-LogReg
This commit is contained in:
+5
-4
@@ -91,22 +91,23 @@ def load_data(filepath):
|
|||||||
- directory (str): 数据文件路径。
|
- directory (str): 数据文件路径。
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
- Tuple[np.array, np.array]: 特征数据和标签数据。
|
- np.array: 特征数据。
|
||||||
|
- np.array: 标签数据。
|
||||||
"""
|
"""
|
||||||
data = []
|
data = []
|
||||||
with open(filepath, 'r') as f:
|
with open(filepath, 'r') as f:
|
||||||
for line in f:
|
for line in f:
|
||||||
row = []
|
row = []
|
||||||
values = line.strip().split()
|
values = line.strip().split()
|
||||||
if '?' in values:
|
|
||||||
continue # Ignore lines with missing data
|
|
||||||
for value in values:
|
for value in values:
|
||||||
|
try:
|
||||||
row.append(float(value))
|
row.append(float(value))
|
||||||
|
except ValueError:
|
||||||
|
row.append(-1)
|
||||||
data.append(row)
|
data.append(row)
|
||||||
np_data = np.array(data)
|
np_data = np.array(data)
|
||||||
X = np_data[:, :-1]
|
X = np_data[:, :-1]
|
||||||
y = np_data[:, -1]
|
y = np_data[:, -1]
|
||||||
y = np.array([0 if label == 1 else (1 if label == 2 else -1) for label in y])
|
|
||||||
return X, y
|
return X, y
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|||||||
Reference in New Issue
Block a user