/house_prices_mlp

Testing out regularized MLPs on the small house prices dataset

Primary LanguagePython

Flax / Jax Practice with Housing Prices

Install dependencies locally, dev-install the package, download data (only needs to be done once).

Get the dataset on Kaggle.

# from repo root
poetry install
# requires kaggle creds
mkdir -p data; cd data
poetry run kaggle competitions download -c house-prices-advanced-regression-techniques
unzip house-prices-advanced-regression-techniques.zip
cp transforms data/

Running tests

poetry run pytest

Interactive jupyter notebooks

poetry run jupyter notebook

Training

poetry run python -m house_prices_mlp.main

TODOs tracker

  • mix out or mix in data augmentation
  • cosine annealing for the learning rate
  • clu logging

Less important / more ambitious

  • ml collections params
  • lifted module with jit?