File size: 522 Bytes
c5c7499
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from model import model
from data_splitting import num_classes, X_train, X_val, input_size
from data_preparing import y_train_encoded, y_val_encoded
from model_callbacks import optimizer, early_stopping, lr_scheduler_callback

model.compile(loss='categorical_crossentropy', optimizer=optimizer, metrics=['accuracy'])

model.fit(X_train, y_train_encoded, epochs=10, batch_size=32, validation_data=(X_val, y_val_encoded), callbacks=[early_stopping,lr_scheduler_callback])

model.save('full_language_identifcation_model1.h5')