# NNI-pred

**Repository Path**: morepeanuts/nni-pred

## Basic Information

- **Project Name**: NNI-pred
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2026-03-16
- **Last Updated**: 2026-03-23

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# NNI-pred

Predicting neonicotinoid insecticide (NNI) concentrations in soil and water using machine learning.

## Background

Neonicotinoids are widely used insecticides that can persist in the environment. This project builds predictive models to estimate NNI concentrations based on environmental, agricultural, and socio-economic factors.

## Features

- Supports soil-only and water-soil merged datasets
- Three model types: Elastic Net, Random Forest, XGBoost
- Nested spatial cross-validation to prevent data leakage
- SHAP-based feature importance analysis

## Usage

```bash
uv run scripts/train_all.py --cls merged --targets THIA IMI CLO
```