尝试一下:
valid_set = dataset.iloc[-60:,:]
test_set = dataset.iloc[-120:-60]
train_set = dataset.iloc[:-120]
概括:
def split_train_test(dataset,validation_size):
valid = dataset.iloc[-validation_size:,:]
train_test = dataset.iloc[:-validation_size)]
train_length = int(0.63 * len(train_test))
# split into input and outputs
train_X,train_y = train_test.iloc[:train_length,:-1],train_test.iloc[:train_length,-1]
test_X,test_y = train_test.iloc[train_length:,train_test.iloc[train_length:,-1]
valid_X,valid_y = valid.iloc[:,valid.iloc[:,-1]
return train_test,valid,train_X,train_y,test_X,test_y,valid_X,valid_y
您可以将%split rati作为参数传递给函数,而不是像我一样将其硬编码为函数。
本文链接:https://www.f2er.com/3058265.html