更新 function.py

2024-10-16 22:07:54 +08:00
parent ecbc4c2515
commit e06bcb1553
1 changed files with 1 additions and 1 deletions
@@ -165,7 +165,7 @@ if __name__ == "__main__":
        "00000000000000000111100000000000"
    )
    predicted_label = classify(
-        np.array([int(char) for char in test_data]).reshape(1, 1024),
+        np.array([int(char) for char in test_data]).reshape(1, 784),
        train_data,
        train_labels,
    )