metadata
license: unknown
metrics:
- accuracy
pipeline_tag: image-classification
tags:
- fin
- image
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>)