Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
xxadev
tensorflow
提交
a9e87682
T
tensorflow
项目概览
xxadev
/
tensorflow
与 Fork 源项目一致
从无法访问的项目Fork
通知
3
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
T
tensorflow
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
a9e87682
编写于
7月 11, 2016
作者:
A
A. Unique TensorFlower
提交者:
TensorFlower Gardener
7月 11, 2016
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Support weighted sparse column in sdca optimizer.
Change: 127093435
上级
cd2162b3
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
38 addition
and
0 deletion
+38
-0
tensorflow/contrib/learn/python/learn/estimators/linear_test.py
...flow/contrib/learn/python/learn/estimators/linear_test.py
+27
-0
tensorflow/contrib/learn/python/learn/estimators/sdca_optimizer.py
...w/contrib/learn/python/learn/estimators/sdca_optimizer.py
+11
-0
未找到文件。
tensorflow/contrib/learn/python/learn/estimators/linear_test.py
浏览文件 @
a9e87682
...
...
@@ -203,6 +203,33 @@ class LinearClassifierTest(tf.test.TestCase):
scores
=
classifier
.
evaluate
(
input_fn
=
input_fn
,
steps
=
2
)
self
.
assertGreater
(
scores
[
'accuracy'
],
0.9
)
def
testSdcaOptimizerWeightedSparseFeatures
(
self
):
"""LinearClasssifier with SDCAOptimizer and weighted sparse features."""
def
input_fn
():
return
{
'example_id'
:
tf
.
constant
([
'1'
,
'2'
,
'3'
]),
'price'
:
tf
.
SparseTensor
(
values
=
[
2.
,
3.
,
1.
],
indices
=
[[
0
,
0
],
[
1
,
0
],
[
2
,
0
]],
shape
=
[
3
,
5
]),
'country'
:
tf
.
SparseTensor
(
values
=
[
'IT'
,
'US'
,
'GB'
],
indices
=
[[
0
,
0
],
[
1
,
0
],
[
2
,
0
]],
shape
=
[
3
,
5
])
},
tf
.
constant
([[
1
],
[
0
],
[
1
]])
country
=
tf
.
contrib
.
layers
.
sparse_column_with_hash_bucket
(
'country'
,
hash_bucket_size
=
5
)
country_weighted_by_price
=
tf
.
contrib
.
layers
.
weighted_sparse_column
(
country
,
'price'
)
sdca_optimizer
=
tf
.
contrib
.
learn
.
SDCAOptimizer
(
example_id_column
=
'example_id'
)
classifier
=
tf
.
contrib
.
learn
.
LinearClassifier
(
feature_columns
=
[
country_weighted_by_price
],
optimizer
=
sdca_optimizer
)
classifier
.
fit
(
input_fn
=
input_fn
,
steps
=
50
)
scores
=
classifier
.
evaluate
(
input_fn
=
input_fn
,
steps
=
2
)
self
.
assertGreater
(
scores
[
'accuracy'
],
0.9
)
def
testSdcaOptimizerCrossedFeatures
(
self
):
"""Tests LinearClasssifier with SDCAOptimizer and crossed features."""
...
...
tensorflow/contrib/learn/python/learn/estimators/sdca_optimizer.py
浏览文件 @
a9e87682
...
...
@@ -124,6 +124,17 @@ class SDCAOptimizer(object):
column
.
length
)
sparse_features
.
append
(
math_ops
.
to_float
(
sparse_features_tensor
))
sparse_features_weights
.
append
(
columns_to_variables
[
column
][
0
])
elif
isinstance
(
column
,
layers
.
feature_column
.
_WeightedSparseColumn
):
# pylint: disable=protected-access
id_tensor
=
column
.
id_tensor
(
transformed_tensor
)
weight_tensor
=
column
.
weight_tensor
(
transformed_tensor
)
sparse_features_tensor
=
sparse_ops
.
sparse_merge
(
id_tensor
,
weight_tensor
,
column
.
length
,
name
=
"{}_sparse_merge"
.
format
(
column
.
name
))
sparse_features
.
append
(
math_ops
.
to_float
(
sparse_features_tensor
,
name
=
"{}_to_float"
.
format
(
column
.
name
)))
sparse_features_weights
.
append
(
columns_to_variables
[
column
][
0
])
else
:
raise
ValueError
(
"SDCAOptimizer does not support column type %s."
%
type
(
column
).
__name__
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录