A financial model package trained on data from Taiwan fiscal year 1999-2009
Project description
Financial Model Taiwan
Financial Model Taiwan is a Python package designed to preprocess, train, and predict financial models. It includes functionalities for data ingestion, preprocessing, resampling, model training, evaluation, and optimization. This package aims to provide a comprehensive solution for financial modeling with support for various machine learning algorithms and techniques.
Model Architecture
The following image illustrates the model architecture:
Workflow Overview
-
Data Ingestion:
- Load data from CSV files.
- Split data into training and testing sets.
-
Preprocessing Pipeline:
- Define preprocessing steps.
- Handle missing values.
- Standardize/normalize data.
- Select important features.
-
Resampling:
- Perform data resampling to handle class imbalance.
-
Initial Model Training:
- Train multiple models: Random Forest, Logistic Regression, SVM, Gradient Boosting, XGBoost, KSBBoost, ANN.
- Evaluate models based on performance metrics.
- Select the best performing models for stacking.
-
Optimizing Model for Performance:
- Stack the best models (XGBoost and Random Forest).
- Optimize hyperparameters using Optuna.
- Tune the stacked model.
- Adjust thresholds to minimize type I and type II errors.
- Increase recall and finalize the model.
Installation
You can install the package via pip:
pip install financial_model_taiwan
Usage
Training a New Model
from financial_model_taiwan import FinModel
pipeline = FinModel(data_path='data/train_data.csv', target_column='target')
pipeline.data_ingestion()
pipeline.data_preprocessing()
pipeline.data_resampling()
pipeline.train_model()
pipeline.save_model('models/trained_model.bin')
evaluation_results = pipeline.evaluate_model()
print(evaluation_results)
Using a Pre-trained Model
from financial_model_taiwan import FinModel
pipeline = FinModel(data_path='data/train_data.csv', target_column='target', model_path='models/trained_model.bin')
pipeline.data_ingestion()
pipeline.data_preprocessing()
pipeline.load_model()
evaluation_results = pipeline.evaluate_model()
print(evaluation_results)
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for financial_model_taiwan-1.0.1.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 533999661c7dbe6e53a1c16e443eca0aaa2965f33419ca79d271ba09972d7d59 |
|
MD5 | 91c5f120853d29916e66ee513d7de55d |
|
BLAKE2b-256 | aa93a88d87b55819e349c2b8efae5eb7ed23a75e679390660fc1bd1a346acda5 |
Hashes for financial_model_taiwan-1.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ef5c89aa5ea95f6c32bf57fe05dc11c3aa3a4d8331d2966e63045b089dca9073 |
|
MD5 | e978a02ed7322a4681a77b947ae67e3c |
|
BLAKE2b-256 | b4763ee28953aa695cffeef4d26967a6779339583867dc665f16caa554827fbd |