text_plain_output_1.png
import matplotlib.pyplot as plt
import os
from collections import Counter
from sklearn import preprocessing
import torch
from torch import nn
from torchvision import transforms, datasets