Commit 368adeb3 authored by wxchan's avatar wxchan Committed by Guolin Ke
Browse files

fix python warning (#342)

parent b7e5f071
...@@ -11,10 +11,10 @@ df_test = pd.read_csv('../binary_classification/binary.test', header=None, sep=' ...@@ -11,10 +11,10 @@ df_test = pd.read_csv('../binary_classification/binary.test', header=None, sep='
W_train = pd.read_csv('../binary_classification/binary.train.weight', header=None)[0] W_train = pd.read_csv('../binary_classification/binary.train.weight', header=None)[0]
W_test = pd.read_csv('../binary_classification/binary.test.weight', header=None)[0] W_test = pd.read_csv('../binary_classification/binary.test.weight', header=None)[0]
y_train = df_train[0] y_train = df_train[0].values
y_test = df_test[0] y_test = df_test[0].values
X_train = df_train.drop(0, axis=1) X_train = df_train.drop(0, axis=1).values
X_test = df_test.drop(0, axis=1) X_test = df_test.drop(0, axis=1).values
num_train, num_feature = X_train.shape num_train, num_feature = X_train.shape
......
...@@ -13,10 +13,10 @@ print('Load data...') ...@@ -13,10 +13,10 @@ print('Load data...')
df_train = pd.read_csv('../regression/regression.train', header=None, sep='\t') df_train = pd.read_csv('../regression/regression.train', header=None, sep='\t')
df_test = pd.read_csv('../regression/regression.test', header=None, sep='\t') df_test = pd.read_csv('../regression/regression.test', header=None, sep='\t')
y_train = df_train[0] y_train = df_train[0].values
X_train = df_train.drop(0, axis=1) y_test = df_test[0].values
y_test = df_test[0] X_train = df_train.drop(0, axis=1).values
X_test = df_test.drop(0, axis=1) X_test = df_test.drop(0, axis=1).values
# create dataset for lightgbm # create dataset for lightgbm
lgb_train = lgb.Dataset(X_train, y_train) lgb_train = lgb.Dataset(X_train, y_train)
......
...@@ -10,10 +10,10 @@ print('Load data...') ...@@ -10,10 +10,10 @@ print('Load data...')
df_train = pd.read_csv('../regression/regression.train', header=None, sep='\t') df_train = pd.read_csv('../regression/regression.train', header=None, sep='\t')
df_test = pd.read_csv('../regression/regression.test', header=None, sep='\t') df_test = pd.read_csv('../regression/regression.test', header=None, sep='\t')
y_train = df_train[0] y_train = df_train[0].values
y_test = df_test[0] y_test = df_test[0].values
X_train = df_train.drop(0, axis=1) X_train = df_train.drop(0, axis=1).values
X_test = df_test.drop(0, axis=1) X_test = df_test.drop(0, axis=1).values
# create dataset for lightgbm # create dataset for lightgbm
lgb_train = lgb.Dataset(X_train, y_train) lgb_train = lgb.Dataset(X_train, y_train)
......
...@@ -10,10 +10,10 @@ print('Load data...') ...@@ -10,10 +10,10 @@ print('Load data...')
df_train = pd.read_csv('../regression/regression.train', header=None, sep='\t') df_train = pd.read_csv('../regression/regression.train', header=None, sep='\t')
df_test = pd.read_csv('../regression/regression.test', header=None, sep='\t') df_test = pd.read_csv('../regression/regression.test', header=None, sep='\t')
y_train = df_train[0] y_train = df_train[0].values
y_test = df_test[0] y_test = df_test[0].values
X_train = df_train.drop(0, axis=1) X_train = df_train.drop(0, axis=1).values
X_test = df_test.drop(0, axis=1) X_test = df_test.drop(0, axis=1).values
print('Start training...') print('Start training...')
# train # train
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment