Advertisement
Guest User

Untitled

a guest
Dec 11th, 2018
59
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.79 KB | None | 0 0
  1. def algo(per_noisy = 0.3, per_Vclean = 0.5):
  2. iris = load_iris()
  3. X_train, X_test, y_train, y_test = train_test_split(iris.data, iris.target, random_state=42)
  4.  
  5. # Bruitage des per_noisy etiquettes
  6. N = len(y_train)
  7. nb_noisy = int(N * per_noisy)
  8. for i in range(nb_noisy):
  9. r = random.randrange(3)
  10. while y_train[i] == r:
  11. r = random.randrange(3)
  12. y_train[i] = r
  13.  
  14. # Création des listes
  15. nb_Vclean = int((N - nb_noisy) * per_Vclean)
  16. Tnoisy = [X_train[:nb_noisy], y_train[:nb_noisy]]
  17. Vclean = [X_train[nb_noisy:nb_noisy+nb_Vclean], y_train[nb_noisy:nb_noisy+nb_Vclean]]
  18. Tclean = [X_train[nb_noisy+nb_Vclean:], y_train[nb_noisy+nb_Vclean:]]
  19. Uclean = [X_test, y_test]
  20.  
  21. return Tnoisy, Vclean, Tclean, Uclean
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement