Skip to content
Snippets Groups Projects
Commit c031cdc8 authored by Iván del Pino's avatar Iván del Pino
Browse files

Correct indentation

parent 403c8e7d
No related branches found
No related tags found
No related merge requests found
...@@ -25,28 +25,28 @@ function GATANeuralNetTraining() ...@@ -25,28 +25,28 @@ function GATANeuralNetTraining()
source_classes_cell_array); source_classes_cell_array);
disp('Done!') disp('Done!')
%% %%
% Now we train the classifier % Now we train the classifier
disp('Training a Shallow Neural Network model for classification...') disp('Training a Shallow Neural Network model for classification...')
[nn_model, tr_x, tr_y, training_Pct_Err] = neural_network_classificator_training(training_dataset, ... [nn_model, tr_x, tr_y, training_Pct_Err] = neural_network_classificator_training(training_dataset, ...
desired_features_indices, classes_names, weight_classes, ... desired_features_indices, classes_names, weight_classes, ...
hiddenLayerSize, trainFcn, performFcn); hiddenLayerSize, trainFcn, performFcn);
disp('Done!') disp('Done!')
%% %%
disp('Using NN for predicting classes...') disp('Using NN for predicting classes...')
[nn_classificated_dataset] = use_nn_classificator(nn_model, ... [nn_classificated_dataset] = use_nn_classificator(nn_model, ...
testing_dataset, desired_features_indices); testing_dataset, desired_features_indices);
disp('Done!') disp('Done!')
%% Now we evaluate the system performance %% Now we evaluate the system performance
disp('Computing performance statistics...') disp('Computing performance statistics...')
[precission, recall, f1_score, overall_precission, ... [precission, recall, f1_score, overall_precission, ...
overall_recall, overall_f1_score, IoU] = ... overall_recall, overall_f1_score, IoU] = ...
evaluating_segmentation_results(nn_classificated_dataset); evaluating_segmentation_results(nn_classificated_dataset);
disp('Done!') disp('Done!')
%% Finally we export the model weights to csv file to use with GATA in ROS %% Finally we export the model weights to csv file to use with GATA in ROS
save_model_to_csv(nn_model, neural_net_filename); save_model_to_csv(nn_model, neural_net_filename);
end end
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment