text_plain_output_1.png
import os

import numpy as np
import pandas as pd
import torch
from torch import nn
import torch.nn.functional as F
from torch import optim
import sklearn
import os
print(os.listdir('../input'))