Environment
Our AstroNet is only implemented in PyTorch 1.4.0 (or higher).
You can install the this package using pypi, as follows:
pip install torchvision pip install torch
Content (Folders):
1. Data: includes process_data.py to handle data;
2. AstroNet_Optimal_AN: includes train.py, model.py, layer.py, utils.py
3. Using_AN_to_ Guide_NN: includes test.py, model.py, layer.py, utils.py;
Run
We directly run process_data.py, train.py and test.py in above files, as follows:
python process_data.py
python train.py
python test.py