Skip to content
Snippets Groups Projects
Commit 2ae48d82 authored by thibault.capt's avatar thibault.capt
Browse files

update wrong dimensions

parent 7b43ffed
Branches
No related tags found
No related merge requests found
...@@ -6,5 +6,5 @@ ...@@ -6,5 +6,5 @@
<component name="MarkdownSettingsMigration"> <component name="MarkdownSettingsMigration">
<option name="stateVersion" value="1" /> <option name="stateVersion" value="1" />
</component> </component>
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.9" project-jdk-type="Python SDK" /> <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.9 (tp-clustering)" project-jdk-type="Python SDK" />
</project> </project>
\ No newline at end of file
...@@ -35,7 +35,7 @@ if __name__ == '__main__': ...@@ -35,7 +35,7 @@ if __name__ == '__main__':
y = dataset.iloc[:, 0].values y = dataset.iloc[:, 0].values
num_features = X.shape[1] num_features = X.shape[1]
learning_rate = 1e-2 learning_rate = 1e-3
max_iterations = 2000 max_iterations = 2000
num_hidden = 10 num_hidden = 10
...@@ -82,12 +82,12 @@ if __name__ == '__main__': ...@@ -82,12 +82,12 @@ if __name__ == '__main__':
print(f"Taux de classifications correctes: {accuracy * 100}%") print(f"Taux de classifications correctes: {accuracy * 100}%")
# Affichage de la droite de séparation des classes # Affichage de la droite de séparation des classes
w1, w2, b = weights_hidden_output[1], weights_hidden_output[2], weights_hidden_output[0] w1, w2, b = weights_input_hidden[1, 1], weights_input_hidden[2, 1], weights_input_hidden[0, 1]
pente = -w1 / w2 pente = -w1 / w2
intercept = -b / w2 intercept = -b / w2
print(f"Droite de séparation: y = {pente}x + {intercept}") print(f"Droite de séparation: y = {pente}x + {intercept}")
# Tracer la droite de séparation (diagonale) # Tracer la droite de séparation
plt.figure() plt.figure()
plt.scatter(X[y == 0][:, 0], X[y == 0][:, 1], color='red', label='Classe 0') plt.scatter(X[y == 0][:, 0], X[y == 0][:, 1], color='red', label='Classe 0')
plt.scatter(X[y == 1][:, 0], X[y == 1][:, 1], color='blue', label='Classe 1') plt.scatter(X[y == 1][:, 0], X[y == 1][:, 1], color='blue', label='Classe 1')
...@@ -97,3 +97,4 @@ if __name__ == '__main__': ...@@ -97,3 +97,4 @@ if __name__ == '__main__':
plt.ylabel('Norm_Grade_2') plt.ylabel('Norm_Grade_2')
plt.legend() plt.legend()
plt.show() plt.show()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment