一直想用 tensonflow,就用一个简单的神经网络(DNN)实验了一下流失预测
方法一:用 contrib.learn 中 DNNClassifier函数
df1 = pd.read_table('E:\\kaggle\\tensonflow\\chunrn_test3', sep=',', encoding="UTF-8")
# print(df1[:5])
print(df1[s_v1][:5])
df1[s_v1] = df1[s_v1].astype(float)
df1['target'] = df1['target'].astype(int)
# scaler = StandardScaler().fit(x_train)
# x_train = scaler.transform(x_train)
df1_train = df1
df1_test = df1
def train_input_fn():
return input_fn(df1_train)
def test_input_fn():
return input_fn(df1_test)
G = globals()
deep_columns = []
for k in s_v1:
temp = tf.contrib.layers.real_valued_column(k)
G[k] = temp
deep_columns.append(G[k])