-
Simple TCN
-
TCN with Residual Blocks
Both models can be trained either using weighted loss or not. We also implemented random prediction as a simple baseline.
Please go to CodeRepo/MODEL_NAME/train_config.py
to change the training configuration. Then run CodeRepo/MODEL_NAME/train.py
The code supports wandb.ai
to monitor the training procedure online.
We tuned hyperparameters on simple TCN. Please run CodeRepo/simple_tcn/hyperpara_opt.py
.
Please go to CodeRepo/MODEL_NAME/test_config.py
to change the testing configuration. Then run CodeRepo/MODEL_NAME/predict.py
. It will save the confusion matrix and metrics (i.e., f1, precision and recall per crop type) as .csv
file.
Please go to CodeRepo/MODEL_NAME/test_config.py
to change the testing configuration. Then run CodeRepo/MODEL_NAME/predict_map.py
.
Please go to CodeRepo/utils/visualize_tf_pred.py
The analysis can be found in ResultEvaluation.ipynb