import pandas as pd
import numpy as np
from sklearn.model_selection import StratifiedKFold
from scipy.stats import rankdata
import lightgbm as lgb
from sklearn import metrics
import gc
from sklearn.model_selection import train_test_split
import warnings
def get_the_vaid_data(train, test):
traind, val = train_test_split(train, test_size=0.2, random_state=1)
train_y = traind['target']
val_y = val['target']
train_x = traind.drop(['target', 'ID_code'], axis=1)
val_x = val.drop(['target', 'ID_code'], axis=1)
test_id = test.ID_code
test_x = test.drop(['ID_code'], axis=1)
return train_x, train_y, val_x, val_y, test_x, test_id
pd.set_option('display.max_columns', 200)
train = pd.read_csv('train.csv')
test = pd.read_csv('test.csv')
#train = train.drop(['var_3', 'var_4&
使用贝叶斯方法自动调参,lgb版本
最新推荐文章于 2024-05-16 12:01:44 发布