-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
- Loading branch information
There are no files selected for viewing
This file was deleted.
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Large diffs are not rendered by default.
This file was deleted.
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,3 @@ | ||
[server-login] | ||
username:tuora | ||
password:Trex1000 |
This file was deleted.
This file was deleted.
This file was deleted.
This file was deleted.
This file was deleted.
This file was deleted.
This file was deleted.
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
out linear(1, True) | ||
-h1 residual_dnn($layers, activation=$activation, skiplayers=2, bn=$bn, keep_prob=$keep_prob) | ||
--x concat($concat_size) | ||
---huser lookup(dataname='user', initrange=$initrange, shape=[None, $uembed]) | ||
---hitem lookup(dataname='item', initrange=$initrange, shape=[None, $iembed]) |
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
out linear(1, True) | ||
-h1 dnn($layers, activation='tanhlecun', bn=True, keep_prob=.95) | ||
--x concat(24) | ||
---huser lookup(dataname='user', initrange=$initrange, shape=[None, $embed]) | ||
---hitem lookup(dataname='item', initrange=$initrange, shape=[None, $embed]) |
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,57 @@ | ||
import tensorflow as tf | ||
from antk.core import config | ||
from antk.core import generic_model | ||
|
||
def dnn_concat(data, configfile, | ||
layers=[16, 8], | ||
activation='tanhlecun', | ||
initrange=1e-3, | ||
bn=True, | ||
keep_prob=.95, | ||
concat_size=24, | ||
uembed=32, | ||
iembed=32, | ||
learnrate=.00001, | ||
verbose=True, | ||
epochs=10, | ||
maxbadcount=20, | ||
mb=2000, | ||
eval_rate=500): | ||
|
||
with tf.name_scope('ant_graph'): | ||
ant = config.AntGraph(configfile, | ||
data=data.dev.features, | ||
marker='-', | ||
graph_name='dnn_concat', | ||
variable_bindings={'layers': layers, | ||
'activation': activation, | ||
'initrange': initrange, | ||
'bn': bn, | ||
'keep_prob': keep_prob, | ||
'concat_size': concat_size, | ||
'uembed': uembed, | ||
'iembed': iembed, | ||
}) | ||
|
||
y = ant.tensor_out | ||
y_ = tf.placeholder("float", [None, None], name='Target') | ||
ant.placeholderdict['ratings'] = y_ # put the new placeholder in the graph for training | ||
with tf.name_scope('objective'): | ||
objective = tf.reduce_sum(tf.square(y_ - y)) | ||
with tf.name_scope('dev_rmse'): | ||
dev_rmse = tf.sqrt(tf.div(tf.reduce_sum(tf.square(y - y_)), data.dev.num_examples)) | ||
with tf.name_scope('training'): | ||
model = generic_model.Model(objective, ant.placeholderdict, | ||
mb=mb, | ||
learnrate=learnrate, | ||
verbose=verbose, | ||
maxbadcount=maxbadcount, | ||
epochs=epochs, | ||
evaluate=dev_rmse, | ||
predictions=y, | ||
model_name='res_concat', | ||
make_histograms=False, | ||
save=False, | ||
tensorboard=False) | ||
model.train(data.train, dev=data.dev, eval_schedule=eval_rate) | ||
return model |
This file was deleted.