Skip to content

Commit

Permalink
add python
Browse files Browse the repository at this point in the history
  • Loading branch information
@Arthur-Null
Arthur-Null committed Feb 10, 2018
1 parent d113474 commit 6e6fdde
Show file tree
Hide file tree
Showing 8 changed files with 14 additions and 14 deletions.
4 changes: 2 additions & 2 deletions AH.py → python/AH.py
View file
Expand Up @@ -189,8 +189,8 @@ def test(beta, omega, num_of_channel, test_data):


num_of_epoches = 20 num_of_epoches = 20


train_path = 'data/train_usr.yzx.txt' train_path = '../data/train_usr.yzx.txt'
test_path = 'data/test_usr.yzx.txt' test_path = '../data/test_usr.yzx.txt'
traindata_size = loadCriteo.count(train_path) traindata_size = loadCriteo.count(train_path)
testdata_size = loadCriteo.count(test_path) testdata_size = loadCriteo.count(test_path)


Expand Down
6 changes: 3 additions & 3 deletions AMTA.py → python/AMTA.py
View file
Expand Up @@ -343,8 +343,8 @@ def vertical_attr(self, lenth):




f_log = open("log_AMTA.txt", 'w') f_log = open("log_AMTA.txt", 'w')
train_path = 'data/train_usr.yzx.txt' train_path = '../data/train_usr.yzx.txt'
test_path = 'data/test_usr.yzx.txt' test_path = '../data/test_usr.yzx.txt'
traindata_size = loadCriteo.count(train_path) traindata_size = loadCriteo.count(train_path)
testdata_size = loadCriteo.count(test_path) testdata_size = loadCriteo.count(test_path)
num_feat = 5867 num_feat = 5867
Expand Down Expand Up @@ -471,6 +471,6 @@ def vertical_attr(self, lenth):
exit(1) exit(1)
f_train = open(train_path) f_train = open(train_path)
f_test = open(test_path) f_test = open(test_path)
AMTAModel = AMTA('./Model/AMTA', batchsize=1000, learning_rate=learning_rate) AMTAModel = AMTA('../Model/AMTA', batchsize=1000, learning_rate=learning_rate)
AMTAModel.train_all_epoch() AMTAModel.train_all_epoch()
AMTAModel.test() AMTAModel.test()
6 changes: 3 additions & 3 deletions ARNN.py → python/ARNN.py
View file
Expand Up @@ -390,8 +390,8 @@ def vertical_attr(self, lenth):




if __name__ == '__main__': if __name__ == '__main__':
traindata = 'data/train_usr.yzx.txt' traindata = '../data/train_usr.yzx.txt'
testdata = 'data/test_usr.yzx.txt' testdata = '../data/test_usr.yzx.txt'
if len(sys.argv) != 4 and len(sys.argv) != 1: if len(sys.argv) != 4 and len(sys.argv) != 1:
print('usage: python ARNN.py [learning rate] [batch size] [mu]') print('usage: python ARNN.py [learning rate] [batch size] [mu]')
exit(1) exit(1)
Expand All @@ -407,7 +407,7 @@ def vertical_attr(self, lenth):
C = config(max_features=5897, learning_rate=learning_rate, batch_size=batch_size, feature_number=12, C = config(max_features=5897, learning_rate=learning_rate, batch_size=batch_size, feature_number=12,
seq_max_len=20, n_input=2, seq_max_len=20, n_input=2,
embedding_output=256, n_hidden=512, n_classes=2, n_epochs=50, isseq=True, miu=mu) embedding_output=256, n_hidden=512, n_classes=2, n_epochs=50, isseq=True, miu=mu)
path = './Model/ARNN' path = '../Model/ARNN'
model = RnnWithattention(path, traindata, testdata, C) model = RnnWithattention(path, traindata, testdata, C)
model.train_until_cov() model.train_until_cov()
model.test(0) model.test(0)
6 changes: 3 additions & 3 deletions DARNN.py → python/DARNN.py
View file
Expand Up @@ -623,8 +623,8 @@ def vertical_attr(self, lenth):




if __name__ == '__main__': if __name__ == '__main__':
traindata = 'data/train_usr.yzx.txt' traindata = '../data/train_usr.yzx.txt'
testdata = 'data/test_usr.yzx.txt' testdata = '../data/test_usr.yzx.txt'
if len(sys.argv) != 4 and len(sys.argv) != 1: if len(sys.argv) != 4 and len(sys.argv) != 1:
print('usage: python ARNN.py [learning rate] [batch size] [mu]') print('usage: python ARNN.py [learning rate] [batch size] [mu]')
exit(1) exit(1)
Expand All @@ -640,7 +640,7 @@ def vertical_attr(self, lenth):
C = config(max_features=5897, learning_rate=learning_rate, batch_size=batch_size, feature_number=12, C = config(max_features=5897, learning_rate=learning_rate, batch_size=batch_size, feature_number=12,
seq_max_len=20, n_input=2, seq_max_len=20, n_input=2,
embedding_output=256, n_hidden=512, n_classes=2, n_epochs=50, isseq=True, miu=mu) embedding_output=256, n_hidden=512, n_classes=2, n_epochs=50, isseq=True, miu=mu)
path = './Model/DARNN' path = '../Model/DARNN'
model = DualAttention(path, traindata, testdata, C) model = DualAttention(path, traindata, testdata, C)
model.train_until_cov() model.train_until_cov()
model.test(0) model.test(0)
6 changes: 3 additions & 3 deletions LR.py → python/LR.py
View file
Expand Up @@ -100,7 +100,7 @@ def train_one_epoch(self):
total_loss = [] total_loss = []
pred = [] pred = []
label = [] label = []
trainfile = open('data/train_usr.yzx.txt', 'rb') trainfile = open('../data/train_usr.yzx.txt', 'rb')
while True: while True:
train_X, train_Y = loadLRF(500, 20, 12, trainfile) train_X, train_Y = loadLRF(500, 20, 12, trainfile)
feed_dict = { feed_dict = {
Expand Down Expand Up @@ -167,7 +167,7 @@ def test(self, epoch):
total_loss = [] total_loss = []
pred = [] pred = []
label = [] label = []
file = open('data/test_usr.yzx.txt', 'rb') file = open('../data/test_usr.yzx.txt', 'rb')
while True: while True:
train_X, train_Y = loadLRF(500, 20, 12, file) train_X, train_Y = loadLRF(500, 20, 12, file)
feed_dict = { feed_dict = {
Expand Down Expand Up @@ -230,6 +230,6 @@ def attr(self):
else: else:
print 'usage: python LR.py [learning rate]' print 'usage: python LR.py [learning rate]'
exit(1) exit(1)
model = LR_f_criteo("./Model/LR", learning_rate=learning_rate) model = LR_f_criteo("../Model/LR", learning_rate=learning_rate)
model.train_all_epochs() model.train_all_epochs()
model.test(0) model.test(0)
0 loadCriteo.py → python/loadCriteo.py
View file
File renamed without changes.
0 model_config.py → python/model_config.py
View file
File renamed without changes.
0 wrapped_loadCriteo.py → python/wrapped_loadCriteo.py
View file
File renamed without changes.

0 comments on commit 6e6fdde

Please sign in to comment.