Commit 21efd467 authored by Shawn You's avatar Shawn You 💬

update

parent 7720cec8
......@@ -28,6 +28,9 @@ predictDataPath = '_3_2_predict-data/data.json'
# 评估数据结果
resultDataPath = '_5_evaluation-result/data.json'
# 训练数据与测试数据比
ratio=4/5
#=======================================
# 选取的特征
#=======================================
......
......@@ -17,7 +17,7 @@ df = df.reset_index(level=None, drop=True).drop(['index'], 1, inplace=False)
totalRowCount = len(df.index)
print('total row count:' + str(totalRowCount))
trainCount = int(totalRowCount * (2/3))
trainCount = int(totalRowCount * _0_config.ratio)
print('train count:' + str(trainCount))
testCount = totalRowCount - trainCount
......
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