# Creation for weight matrices for initialising the first layer of a neural network

## Dependencies & Environment

- Conda needs to be installed
- Setup the conda environment `weightmatrices` by running:

    ```
    bash setup_dependencies.sh
    ```

## Usage

- To run the main script(s) that produce the weight matrices run:

    ```
    bash run.sh
    ```

    If you want to directly run `main.py` to use command line args (e.g. `python main.py --methods 'pca' --nhidden 10 25`), do not forget to activate the environment by running:

    ```
    conda activate weightmatrices
    ```

## Remarks

By default, Datasets are downloaded and put into a /datasets directory.  
To avoid this you can:
1. Change the root variable in the torchvision data loader function.
2. Create the /datasets folder and corresponding subfolder and put the datasets (or a link) in there.