python data_Python data.Data方法代码示例

本文整理汇总了Python中data.Data方法的典型用法代码示例。如果您正苦于以下问题：Python data.Data方法的具体用法？Python data.Data怎么用？Python data.Data使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在模块data的用法示例。

在下文中一共展示了data.Data方法的11个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于我们的系统推荐出更棒的Python代码示例。

示例1: __init__

​点赞 6

​

# 需要导入模块: import data [as 别名]

# 或者: from data import Data [as 别名]

def __init__(self):

super(Predictor, self).__init__()

num_units = 512

num_layer = 2

batch_size = 1

data_dir = 'data/'

input_file = 'poetry.txt'

vocab_file = 'vocab.pkl'

tensor_file = 'tensor.npy'

self.data = Data(data_dir, input_file, vocab_file, tensor_file,

is_train=False, batch_size=batch_size)

self.model = Net(self.data, num_units, num_layer, batch_size)

self.sess = tf.Session()

saver = tf.train.Saver(tf.global_variables())

saver.restore(self.sess, 'model/model')

print('Load model done.' + '\n')

开发者ID:stardut，项目名称:Text-Generate-RNN，代码行数:20，

示例2: main

​点赞 6

​

# 需要导入模块: import data [as 别名]

# 或者: from data import Data [as 别名]

def main():

global model

if args.data_test == ['video']:

from videotester import VideoTester

model = model.Model(args, checkpoint)

t = VideoTester(args, model, checkpoint)

t.test()

else:

if checkpoint.ok:

loader = data.Data(args)

_model = model.Model(args, checkpoint)

_loss = loss.Loss(args, checkpoint) if not args.test_only else None

t = Trainer(args, loader, _model, _loss, checkpoint)

while not t.terminate():

t.train()

t.test()

checkpoint.done()

开发者ID:thstkdgus35，项目名称:EDSR-PyTorch，代码行数:20，

示例3: test

​点赞 6

​

# 需要导入模块: import data [as 别名]

# 或者: from data import Data [as 别名]

def test():

from data import Data

from config import Config

conf = Config()

usecuda = True

we = torch.load('./data/processed/ji/we.pkl')

char_table = None

sub_table = None

if conf.need_char or conf.need_elmo:

char_table = torch.load('./data/processed/ji/char_table.pkl')

if conf.need_sub:

sub_table = torch.load('./data/processed/ji/sub_table