Skip to content

Instantly share code, notes, and snippets.

@piEsposito

piEsposito/blitz_example_imports.py

Created March 30, 2020 18:54
Show Gist options
  • Save piEsposito/b40d23da6a5119c6758dbae91515707a to your computer and use it in GitHub Desktop.
Save piEsposito/b40d23da6a5119c6758dbae91515707a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
blitz_example_imports.py
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
import numpy as np
from blitz.modules import BayesianLinear
from blitz.utils import variational_estimator
from sklearn.datasets import load_boston
from sklearn.preprocessing import StandardScaler
from sklearn.model_selection import train_test_split
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment