learning o 说明文档
概述工具函数
功能
类/函数
描述
计算类别相似度
get_rehearsal_prototype()
根据 memory 和 memory_o 数据集预测的特征和标签计算每个类别的余弦相似度
预测特征
get_token_features_and_labels()
使用上一轮的模型预测特征
预测隐藏状态序列
get_token_encodings_and_labels()
使用上一轮的模型预测最后一层输出端的隐藏状态序列
计算类别的原型
get_exemplar_means()
根据支持集的 encoding 计算每个类别的原型,即均值
准备数据集
功能
类/函数
描述
表示训练/测试样本的对象
InputExample(object)
包含每个样本的编号,单词序列,标签序列
表示样本特征集的对象
InputFeatures(object)
包含每个样本特征的 input_ids,input_mask,segment_ids,label_ids
从文件中读取数据
read_examples_from_file( ...