tune with optuna and model
Project description
trainme - documentation
Authors
kaggle-autolgb
trainer is a combination of xgboost, lightgbm and optuna. I tried to make kaggle monthly competition simple. Its only working with classification problem.
Installation
pip install trainer
Features
- autotune
- autotrain
- auto submission file generate
- auto prediction
Deployment
from src.read_data import ReadFile
s = ReadFile(
train_path="/home/aditta/Desktop/trainme/trainme/input/multi_class_classification.csv",
test_path="/home/aditta/Desktop/trainme/trainme/input/multi_class_classification_test.csv",
label="target",
task_type="multi_classification",
compare=False,
fold="skfold",
model_name="xgb",
output_path="/media/aditta/NewVolume/amazon",
study_name="new_train",
store_file ="out9",
n_trials=1
)
print(s.report())
print(s.train())
License
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
trainme-1.0.tar.gz
(10.6 kB
view hashes)
Built Distribution
trainme-1.0-py3-none-any.whl
(16.4 kB
view hashes)