[6ab127]: / classification.py

Download this file

16 lines (12 with data), 741 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
# Preprocessing the dataset
dataset_path = 'path/to/dataset'
max_sequence_length = 1000
train_sequences, test_sequences, train_labels, test_labels, label_mapping = preprocess_dataset(dataset_path, max_sequence_length)
# Creating and training the CNN model
cnn_model = create_cnn_model(train_sequences.shape, len(label_mapping))
cnn_accuracy = train_and_evaluate_model(cnn_model, train_sequences, train_labels, test_sequences, test_labels)
# Creating and training the RNN model
rnn_model = create_rnn_model(train_sequences.shape, len(label_mapping))
rnn_accuracy = train_and_evaluate_model(rnn_model, train_sequences, train_labels, test_sequences, test_labels)
print(f"CNN Accuracy: {cnn_accuracy}")
print(f"RNN Accuracy: {rnn_accuracy}")