You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

swish.py 1.3KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. # -*- coding: utf-8 -*-
  2. """
  3. Created on Wed Nov 24 16:48:52 2021
  4. @author: virgi
  5. """
  6. import tensorflow as tf
  7. import matplotlib.pyplot as plt
  8. from fonction_activation import *
  9. from Creation_donnee import *
  10. import numpy as np
  11. n=20
  12. #création de la base de donnéé
  13. X,Y=creation_sin(-15,-8,n,1,)
  14. X2,Y2=creation_sin(10,18,n,1,)
  15. X=np.concatenate([X,X2])
  16. Y=np.concatenate([Y,Y2])
  17. n=10000
  18. Xv,Yv=creation_sin(-20,20,n,1)
  19. model_swish=tf.keras.models.Sequential()
  20. model_swish.add(tf.keras.Input(shape=(1,)))
  21. model_swish.add(tf.keras.layers.Dense(512, activation='swish'))
  22. model_swish.add(tf.keras.layers.Dense(1))
  23. opti=tf.keras.optimizers.Adam()
  24. model_swish.compile(opti, loss='mse', metrics=['accuracy'])
  25. model_swish.summary()
  26. model_swish.fit(X, Y, batch_size=1, epochs=10, shuffle='True',validation_data=(Xv, Yv))
  27. Y_predis_swish=model_swish.predict(X)
  28. Y_predis_validation_swish=model_swish.predict(Xv)
  29. plt.figure()
  30. plt.plot(X,Y,'x',label='donnée')
  31. plt.plot(Xv,Yv,label="validation")
  32. plt.plot(X,Y_predis_swish,'o',label='prediction sur les donné avec swish comme activation')
  33. plt.plot(Xv,Y_predis_validation_swish,label='prediction sur la validation avec swish comme activation')
  34. plt.legend()
  35. plt.show()
  36. """
  37. Created on Wed Nov 24 16:53:37 2021
  38. @author: virgi
  39. """