Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
曾经的那一瞬间
Models
提交
21b48a85
M
Models
项目概览
曾经的那一瞬间
/
Models
11 个月 前同步成功
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
Models
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
21b48a85
编写于
11月 07, 2017
作者:
N
Neal Wu
提交者:
GitHub
11月 07, 2017
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #2718 from tensorflow/flaky-wide-deep-test
Make wide_deep_test much less flaky and more thorough
上级
205c5e07
dbd3088c
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
18 addition
and
3 deletion
+18
-3
official/wide_deep/wide_deep_test.csv
official/wide_deep/wide_deep_test.csv
+10
-0
official/wide_deep/wide_deep_test.py
official/wide_deep/wide_deep_test.py
+8
-3
未找到文件。
official/wide_deep/wide_deep_test.csv
浏览文件 @
21b48a85
...
...
@@ -18,3 +18,13 @@
32,Private,186824,HS-grad,9,Never-married,Machine-op-inspct,Unmarried,,,0,0,40,,<=50K
38,Private,28887,11th,7,Married-civ-spouse,Sales,Husband,,,0,0,50,,<=50K
43,Self-emp-not-inc,292175,Masters,14,Divorced,Exec-managerial,Unmarried,,,0,0,45,,>50K
40,Private,193524,Doctorate,16,Married-civ-spouse,Prof-specialty,Husband,,,0,0,60,,>50K
56,Local-gov,216851,Bachelors,13,Married-civ-spouse,Tech-support,Husband,,,0,0,40,,>50K
54,?,180211,Some-college,10,Married-civ-spouse,?,Husband,,,0,0,60,,>50K
22,State-gov,311512,Some-college,10,Married-civ-spouse,Other-service,Husband,,,0,0,15,,<=50K
31,Private,84154,Some-college,10,Married-civ-spouse,Sales,Husband,,,0,0,38,,>50K
57,Federal-gov,337895,Bachelors,13,Married-civ-spouse,Prof-specialty,Husband,,,0,0,40,,>50K
47,Private,51835,Prof-school,15,Married-civ-spouse,Prof-specialty,Wife,,,0,1902,60,,>50K
50,Federal-gov,251585,Bachelors,13,Divorced,Exec-managerial,Not-in-family,,,0,0,55,,>50K
25,Private,289980,HS-grad,9,Never-married,Handlers-cleaners,Not-in-family,,,0,0,35,,<=50K
42,Private,116632,Doctorate,16,Married-civ-spouse,Prof-specialty,Husband,,,0,0,45,,>50K
official/wide_deep/wide_deep_test.py
浏览文件 @
21b48a85
...
...
@@ -85,18 +85,23 @@ class BaseTest(tf.test.TestCase):
input_fn
=
lambda
:
wide_deep
.
input_fn
(
TEST_CSV
,
num_epochs
=
1
,
shuffle
=
False
,
batch_size
=
1
))
# Train for
40 steps at batch size 2
and evaluate final loss
# Train for
100 epochs at batch size 3
and evaluate final loss
model
.
train
(
input_fn
=
lambda
:
wide_deep
.
input_fn
(
TEST_CSV
,
num_epochs
=
None
,
shuffle
=
True
,
batch_size
=
2
),
steps
=
40
)
TEST_CSV
,
num_epochs
=
100
,
shuffle
=
True
,
batch_size
=
3
))
final_results
=
model
.
evaluate
(
input_fn
=
lambda
:
wide_deep
.
input_fn
(
TEST_CSV
,
num_epochs
=
1
,
shuffle
=
False
,
batch_size
=
1
))
print
(
'%s initial results:'
%
model_type
,
initial_results
)
print
(
'%s final results:'
%
model_type
,
final_results
)
# Ensure loss has decreased, while accuracy and both AUCs have increased.
self
.
assertLess
(
final_results
[
'loss'
],
initial_results
[
'loss'
])
self
.
assertGreater
(
final_results
[
'auc'
],
initial_results
[
'auc'
])
self
.
assertGreater
(
final_results
[
'auc_precision_recall'
],
initial_results
[
'auc_precision_recall'
])
self
.
assertGreater
(
final_results
[
'accuracy'
],
initial_results
[
'accuracy'
])
def
test_wide_deep_estimator_training
(
self
):
self
.
build_and_test_estimator
(
'wide_deep'
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录