Antrenare cu validare încrucișată

Este momentul să antrenezi modelul cu cei mai buni parametri găsiți: 0,001 pentru rata de învățare, 50 de epoci, un batch_size de 128 și activări relu.

Funcția create_model() din exercițiul anterior este pregătită pentru a fi folosită. X și y sunt încărcate ca atribute și etichete.

Folosește valorile optime găsite pentru model atunci când creezi obiectul KerasClassifier, astfel încât acestea să fie aplicate în timpul validării încrucișate.

Încheie acest capitol antrenând un model bine calibrat pe setul de date despre cancerul de sân!

Importă KerasClassifier din wrappers-urile scikit_learn ale tensorflow.keras.
Creează un obiect KerasClassifier furnizând cei mai buni parametri găsiți.
Transmite model-ul tău, atributele și etichetele către cross_val_score pentru a efectua validarea încrucișată cu 3 fold-uri.

.css-6su6fj{-webkit-flex-shrink:0;-ms-flex-negative:0;flex-shrink:0;}exercițiu

Instrucțiuni

exercițiu