Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in
Toggle navigation
G
gm_strategy_cvr
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
rank
gm_strategy_cvr
Commits
68f0379a
Commit
68f0379a
authored
Jul 22, 2020
by
赵威
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
try speed
parent
e41c6242
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
9 deletions
+4
-9
model.py
src/models/esmm/model.py
+4
-9
No files found.
src/models/esmm/model.py
View file @
68f0379a
...
...
@@ -92,8 +92,7 @@ def _bytes_feature(value):
return
tf
.
train
.
Feature
(
bytes_list
=
tf
.
train
.
BytesList
(
value
=
[
value
]))
@numba.jit
(
parallel
=
True
)
def
_make_example
(
inputs
):
def
model_predict
(
inputs
,
predict_fn
):
time_1
=
timeit
.
default_timer
()
int_columns
=
[
"active_type"
,
"active_days"
,
"card_id"
,
"is_pure_author"
,
"is_have_reply"
,
"is_have_pure_reply"
,
"content_level"
,
...
...
@@ -107,20 +106,16 @@ def _make_example(inputs):
if
col
in
[
"click_label"
,
"conversion_label"
]:
pass
elif
col
in
int_columns
:
features
[
col
]
=
_int64_feature
(
int
(
value
))
features
[
col
]
=
tf
.
train
.
Feature
(
int64_list
=
tf
.
train
.
Int64List
(
value
=
[
int
(
value
)]
))
elif
col
in
float_columns
:
features
[
col
]
=
_float_feature
(
float
(
value
))
features
[
col
]
=
tf
.
train
.
Feature
(
float_list
=
tf
.
train
.
FloatList
(
value
=
[
float
(
value
)]
))
else
:
features
[
col
]
=
_bytes_feature
(
str
(
value
)
.
encode
(
encoding
=
"utf-8"
))
features
[
col
]
=
tf
.
train
.
Feature
(
bytes_list
=
tf
.
train
.
BytesList
(
value
=
[
str
(
value
)
.
encode
(
encoding
=
"utf-8"
)]
))
example
=
tf
.
train
.
Example
(
features
=
tf
.
train
.
Features
(
feature
=
features
))
examples
.
append
(
example
.
SerializeToString
())
total_1
=
(
timeit
.
default_timer
()
-
time_1
)
print
(
"make example cost {:.5f}s"
.
format
(
total_1
))
return
examples
def
model_predict
(
inputs
,
predict_fn
):
examples
=
_make_example
(
inputs
)
time_1
=
timeit
.
default_timer
()
predictions
=
predict_fn
({
"examples"
:
examples
})
total_1
=
(
timeit
.
default_timer
()
-
time_1
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment