Skip to content

Instantly share code, notes, and snippets.

@nilsleh
nilsleh / inpainting.py
Created May 22, 2024 14:22
Inpainting procedure with DDPM and RePaint
from glob import glob
import os
import matplotlib.pyplot as plt
import torch
from denoising_diffusion_pytorch import Unet
from denoising_diffusion_pytorch.repaint import GaussianDiffusion as RePaint
from PIL import Image
from torch.utils.data import DataLoader
from torchvision.datasets import VisionDataset
@nilsleh
nilsleh / min_example.py
Last active April 9, 2024 09:26
DeepSensor Reproducable Example
from typing import Any
import deepsensor.torch
import numpy as np
import xarray as xr
from deepsensor.data.loader import TaskLoader
from deepsensor.data.processor import DataProcessor
from deepsensor.train import Trainer
from hydra.utils import instantiate
from omegaconf import DictConfig, OmegaConf
@nilsleh
nilsleh / laplace_example.py
Last active January 23, 2024 07:44
Laplace Example Constant UQ Reproduce
import os
from functools import lru_cache
from typing import Any, Dict, Optional
import kornia.augmentation as K
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import timm
import torch
@nilsleh
nilsleh / ocean_bench_datamodule.py
Last active January 16, 2024 15:44
Ocean Bench Lightning Datamodule
"""Ocean Bench Datamodules."""
import itertools
import os
from collections import namedtuple
from typing import Any
import hydra
import numpy as np
import ocn_tools._src.geoprocessing.gridding as obgrid
@nilsleh
nilsleh / raps.py
Last active January 11, 2024 15:06
Compare RAPS implementations
import os
from typing import Sequence
import matplotlib.pyplot as plt
import numpy as np
import torch
import torch.backends.cudnn as cudnn
from lightning import Trainer
from lightning.pytorch import seed_everything
from orig_conformal import ConformalModel
@nilsleh
nilsleh / retrieve_milestone_contributors.py
Created November 10, 2023 13:18
Retrieve the contributors from a milestone
import requests
# Set up authentication with your PAT
headers = {
'Authorization': 'Bearer YOUR_ACCESS_TOKEN'
}
# Define the repository information
repo_owner = 'repo_owner'
repo_name = 'repo_name'
@nilsleh
nilsleh / 1d regression_swag.ipynb
Last active February 26, 2023 16:34
SWAG 1D Regression example
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
import torch
import timm
from torchgeo.datasets import EuroSAT
import numpy as np
from sklearn.linear_model import LogisticRegression
from torch.utils.data import DataLoader
import torch
from tqdm import tqdm
import numpy as np
import pandas as pd
from darts import TimeSeries
from darts.dataprocessing.transformers import Scaler
from darts.models import TFTModel
from darts.datasets import AirPassengersDataset
from darts.utils.timeseries_generation import datetime_attribute_timeseries
from darts.utils.likelihood_models import QuantileRegression
import pytorch_lightning as pl
@nilsleh
nilsleh / darts_load_checkpoint.py
Last active May 8, 2024 20:41
Exploring saving and loading checkpoint
import numpy as np
import pandas as pd
from tqdm import tqdm_notebook as tqdm
import matplotlib.pyplot as plt
from darts import TimeSeries, concatenate
from darts.dataprocessing.transformers import Scaler
from darts.models import TFTModel
from darts.metrics import mape