Enric Moreu enric1994

Data Scientist

shaypal5 / confusion_matrix_pretty_print.py

Last active May 13, 2024 13:04

Pretty print a confusion matrix with seaborn

	import pandas as pd
	import matplotlib.pyplot as plt
	import seaborn as sns

	def print_confusion_matrix(confusion_matrix, class_names, figsize = (10,7), fontsize=14):
	"""Prints a confusion matrix, as returned by sklearn.metrics.confusion_matrix, as a heatmap.

	Note that due to returning the created figure object, when this funciton is called in a
	notebook the figure willl be printed twice. To prevent this, either append ; to your
	function call, or modify the function by commenting out the return expression.

jcjohnson / pytorch_finetune.py

Last active June 19, 2024 17:28

	import argparse
	import torch
	import torch.nn as nn
	from torch.autograd import Variable
	from torch.utils.data import DataLoader

	import torchvision
	import torchvision.transforms as T
	from torchvision.datasets import ImageFolder

Nikhil-Kasukurthi / visualisation.py

Last active December 13, 2022 16:55

Visualisation of CNN using Grad-Cam on PyTorch

	import cv2
	import numpy as np
	import torch
	from torchvision import models, transforms
	from torch.autograd import Variable
	import torch
	import torch.nn as nn
	import pickle
	import os
	import argparse

tejaskhot / shapenet_synset_list

Created June 24, 2018 00:44

List of category names and their id in the ShapeNet dataset