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