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
39ac4eb7
Commit
39ac4eb7
authored
Jul 22, 2020
by
赵威
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
try numba
parent
247a7c1d
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
7 deletions
+2
-7
model.py
src/models/esmm/model.py
+2
-7
No files found.
src/models/esmm/model.py
View file @
39ac4eb7
import
timeit
import
numba
import
tensorflow
as
tf
from
tensorflow
import
feature_column
as
fc
from
tensorflow.python.estimator.canned
import
head
as
head_lib
...
...
@@ -94,6 +95,7 @@ def _bytes_feature(value):
return
tf
.
train
.
Feature
(
bytes_list
=
tf
.
train
.
BytesList
(
value
=
[
value
]))
@numba.jit
(
nopython
=
True
,
parallel
=
True
)
def
model_predict2
(
device_id
,
diary_ids
,
device_dict
,
diary_dict
,
predict_fn
):
time_1
=
timeit
.
default_timer
()
device_info
,
diary_lst
=
device_diary_fe
(
device_id
,
diary_ids
,
device_dict
,
diary_dict
)
...
...
@@ -114,13 +116,6 @@ def model_predict2(device_id, diary_ids, device_dict, diary_dict, predict_fn):
tmp
.
update
(
device_info
)
tmp
.
update
(
diary_info
)
features
=
{}
# for (col, value) in tmp.items():
# if col in int_columns:
# features[col] = _int64_feature(int(value))
# elif col in float_columns:
# features[col] = _float_feature(float(value))
# elif col in str_columns:
# features[col] = _bytes_feature(str(value).encode(encoding="utf-8"))
for
col
in
int_columns
:
features
[
col
]
=
_int64_feature
(
int
(
tmp
[
col
]))
for
col
in
float_columns
:
...
...
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