Code Cleanup

[collin-k]

Description
This PR addresses code readability and maintainability.

Changes

1. The config.REGULARIZATION_TYPE and config.REGULARIZATION_WEIGHT in the train_epoch function in train.py are now passed as flat parameters rather than as a tuple.
2. Configuration variables for KaneCounty and RiverDatset classes were revised to be passed as individual parameters arguments rather than as a tuple. The code to instantiate KaneCounty and RiverDataset within train.py (in the initialize_dataset function) was changed accordingly. The docstrings within each class was revised accordingly.
3. Variables within KaneCounty and RiverDataset were changed to instance variable using self. The docstrings within each class was revised accordingly.
4. Configuration variables for BalancedRandomBatchGeoSampler and BalancedGridGeoSampler classes were revised to be passed as individual parameters arguments rather than as a dictionary. The code to instantiate BalancedRandomBatchGeoSampler and BalancedGridGeoSampler within train.py (in the build_dataloader function) was partially changed accordingly (only lines 224-233. The docstrings within each class was revised accordingly.

[annawoodard]

wonderful thank you @collin-k!!!!!