New_T

MLP Classificação

Oct 28th, 2021 (edited)
142
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.36 KB | None | 0 0
  1. import tensorflow as tf
  2. import numpy as np
  3. from sklearn.datasets import load_breast_cancer
  4. from sklearn.model_selection import train_test_split
  5.  
  6. dadosBreastCancer = load_breast_cancer()
  7. entradaDeDadosBreastCancer, saidaDeDadosBreastCancer = dadosBreastCancer.data, dadosBreastCancer.target
  8. entradaDeDadosBreastCancerTreino, entradaDeDadosBreastCancerTeste, saidaDeDadosBreastCancerTreinamento, saidaDeDadosBreastCancerTeste = train_test_split(entradaDeDadosBreastCancer, saidaDeDadosBreastCancer, test_size=0.3, random_state=1)
  9.  
  10. print(entradaDeDadosBreastCancer.shape[0])
  11.  
  12.  
  13. model = tf.keras.models.Sequential([
  14.   tf.keras.Input(shape=(entradaDeDadosBreastCancer.shape[1],)),
  15.   tf.keras.layers.Dense(32, activation='relu'),
  16.   tf.keras.layers.Dropout(0.2),
  17.   tf.keras.layers.Dense(32, activation='relu'),
  18.   tf.keras.layers.Dropout(0.2),
  19.   tf.keras.layers.Dense(2, activation='softmax')
  20. ])
  21. model.compile(optimizer='adam',
  22.               loss='sparse_categorical_crossentropy',
  23.               metrics=['accuracy'])
  24. model.fit(entradaDeDadosBreastCancer, saidaDeDadosBreastCancer, epochs=100)
  25.  
  26. print("Acuracia no treinamento",model.evaluate(entradaDeDadosBreastCancerTreino,  saidaDeDadosBreastCancerTreinamento, verbose=2)              
  27. )
  28.  
  29. print("Acuracia no test", model.evaluate(entradaDeDadosBreastCancerTeste,  saidaDeDadosBreastCancerTeste, verbose=2)              
  30. )
  31.  
Add Comment
Please, Sign In to add comment