Skip to content

Instantly share code, notes, and snippets.

@MartinWeiss12
Last active October 23, 2024 17:32
Show Gist options
  • Save MartinWeiss12/dacdfc090610b84aaa5508a36020261d to your computer and use it in GitHub Desktop.
Save MartinWeiss12/dacdfc090610b84aaa5508a36020261d to your computer and use it in GitHub Desktop.
Imports
import torch
from torch.utils.data import DataLoader
from torchvision import datasets, transforms
import torch.nn as nn
import torch.nn.functional as F
import numpy as np
import cv2
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import torch.optim as optim
from tqdm.notebook import tqdm
import os
import matplotlib.colors as colors
import torch.optim.lr_scheduler as lr_scheduler
import pandas as pd
from sklearn.metrics import precision_score, recall_score, f1_score, roc_auc_score, roc_curve, auc
import time
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment