提交 f8c5cad2 编写于 作者: W wuzewu

Update task API

上级 7b1fc7c8
...@@ -554,10 +554,10 @@ class BasicTask(object): ...@@ -554,10 +554,10 @@ class BasicTask(object):
class ClassifierTask(BasicTask): class ClassifierTask(BasicTask):
def __init__(self, def __init__(self,
data_reader,
feature, feature,
num_classes, num_classes,
feed_list, feed_list,
data_reader,
startup_program=None, startup_program=None,
config=None, config=None,
hidden_units=None): hidden_units=None):
...@@ -662,10 +662,10 @@ ImageClassifierTask = ClassifierTask ...@@ -662,10 +662,10 @@ ImageClassifierTask = ClassifierTask
class TextClassifierTask(ClassifierTask): class TextClassifierTask(ClassifierTask):
def __init__(self, def __init__(self,
data_reader,
feature, feature,
num_classes, num_classes,
feed_list, feed_list,
data_reader,
startup_program=None, startup_program=None,
config=None, config=None,
hidden_units=None): hidden_units=None):
...@@ -711,8 +711,8 @@ class SequenceLabelTask(BasicTask): ...@@ -711,8 +711,8 @@ class SequenceLabelTask(BasicTask):
feature, feature,
max_seq_len, max_seq_len,
num_classes, num_classes,
data_reader,
feed_list, feed_list,
data_reader,
startup_program=None, startup_program=None,
config=None, config=None,
): ):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册