File size: 2,955 Bytes
3f84e4b 65fd3e0 3f84e4b 65fd3e0 3f84e4b 65fd3e0 457a565 65fd3e0 3f84e4b 65fd3e0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 | ---
# For reference on model card metadata, see the spec: https://github.com/huggingface/hub-docs/blob/main/modelcard.md?plain=1
# Doc / guide: https://huggingface.co/docs/hub/model-cards
{}
---
# Model Card for Model ID
The Hybrid Image-Numerical Stock Prediction Model is a multi-input neural network designed to predict binary stock price movements (up/down) based on historical chart images and technical indicators. It processes both image data (stock charts) and numerical data (technical indicators like RSI, MACD, Bollinger Bands, ATR, and OBV) to make predictions.
This modelcard aims to be a base template for new models. It has been generated using [this raw template](https://github.com/huggingface/huggingface_hub/blob/main/src/huggingface_hub/templates/modelcard_template.md?plain=1).
## Uses
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
### Direct Use
The model can be used for predicting short-term stock price movements based on historical data. It is intended for use in financial analysis when combined with other tools and strategies.
## Bias, Risks, and Limitations
The model may exhibit biases based on the training dataset's characteristics. It may not perform well during sudden market shifts caused by external factors. Users should be cautious about over-reliance on the model's predictions without additional analysis.
### Recommendations
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
## How to Get Started with the Model
## Training Details
### Training Data
The model was trained on the Kaylah072001/Stock_Graphs_Dataset, which includes stock chart images and corresponding price data. Technical indicators were extracted using the ta library.
### Training Procedure
<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
#### Preprocessing [optional]
Image data was rescaled to a range of 0–1 and augmented using rotation, shift, zoom, and flip techniques. Numerical features were extracted dynamically during training using technical indicator calculations.
#### Training Hyperparameters
- **Training regime:** Training regime: Mixed precision (fp32)
Optimizer: Adam
Loss function: Binary crossentropy
Metrics: Accuracy, AUC-ROC
Cross-validation: 5-fold stratified k-fold
### Testing Data, Factors & Metrics
#### Testing Data
<!-- This should link to a Dataset Card if possible. -->
[More Information Needed]
#### Factors
Performance was evaluated across different market conditions and stock types.
[More Information Needed]
#### Metrics
Accuracy
|