Abner0803/multiclass-stock-cnn

Download model

git lfs install
git clone https://huggingface.co/Abner0803/multiclass-stock-cnn

Usage

In directory ./multiclass-stock-cnn, you can try the pre-trained model out

touch expr.py

pip install torch pillow numpy

## expr.py

import torch
import numpy as np
from PIL import Image


image_file = "temp.png"
matrix = np.array(Image.open(image_file))
tensor = torch.Tensor(matrix, dtype=torch.float32)

pred = model(tensor.unsqueeze(0).unsqueeze(0))

print(pred)

You'll get the probability of each class

tensor([[9.9691e-01, 1.1429e-09, 3.4367e-20, 6.5349e-30, 9.6665e-25, 1.1544e-27,
         2.8234e-11, 6.8401e-21, 2.1943e-16, 3.0882e-03]],
       grad_fn=<SoftmaxBackward0>)