function to covert mnist to mat in matlab

main.m
function convert_mnist_to_mat(input_filename, output_filename)
    % read the MNIST data from the input filename
    [X_train, y_train] = read_mnist(input_filename);
    
    % create a struct to store the data
    mnist_data = struct('X_train', X_train, 'y_train', y_train);
    
    % save the data to a .mat file
    save(output_filename, 'mnist_data');
end
361 chars
11 lines

main.m
function [X_train, y_train] = read_mnist(filename)
    % open the binary file
    fid = fopen(filename, 'r','b');
    
    % read the header information
    magic = fread(fid, 1, 'int32');
    num_images = fread(fid, 1, 'int32');
    rows = fread(fid, 1, 'int32');
    cols = fread(fid, 1, 'int32');
    
    % read the image data
    X_train = fread(fid, [rows*cols, num_images], 'uint8');
    X_train = double(X_train') / 255;
    
    % read the label data
    magic = fread(fid, 1, 'int32');
    num_labels = fread(fid, 1, 'int32');
    y_train = fread(fid, [num_labels, 1], 'uint8');
    
    % close the file
    fclose(fid);
end
636 chars
23 lines

main.m
input_filename = 'train-images-idx3-ubyte';
output_filename = 'mnist_data.mat';
convert_mnist_to_mat(input_filename, output_filename);
135 chars
4 lines