torch4keras.snippets module¶
- class torch4keras.snippets.IterDataset(*args: Any, **kwargs: Any)[source]¶
流式读取文件，用于大数据量、多小文件使用时候需要注意steps_per_epoch != None
- Parameters:
file_path – str, 待读取的文件的路径，若无可以为None
- class torch4keras.snippets.ListDataset(*args: Any, **kwargs: Any)[source]¶
数据是List格式Dataset，支持传入file_path或者外部已读入的data(List格式)
- Parameters:
file_path – str, 待读取的文件的路径，若无可以为None
data – List[Any], list格式的数据，和file_path至少有一个不为None
- torch4keras.snippets.colorful(obj, color='yellow', display_type='plain')[source]¶
# 彩色输出格式： # 设置颜色开始 ：[显示方式;前景色;背景色m # 说明： # 前景色 背景色 颜色 # ————————————— # 30 40 黑色 # 31 41 红色 # 32 42 绿色 # 33 43 黃色 # 34 44 蓝色 # 35 45 紫红色 # 36 46 青蓝色 # 37 47 白色 # 显示方式 意义 # ————————- # 0 终端默认设置 # 1 高亮显示 # 4 使用下划线 # 5 闪烁 # 7 反白显示 # 8 不可见
- torch4keras.snippets.email_when_error(receivers, **configs)[source]¶
装饰器，异常则发邮件 Example: ——– @email_when_error(receivers=’tongjilibo@163.com’) def test():
return 1/0
test() # 调用
- torch4keras.snippets.info_level_prefix(string, level=0)[source]¶
在字符串前面加上有颜色的[INFO][WARNING][ERROR]字样
- torch4keras.snippets.metric_mapping(metric, func, y_pred, y_true)[source]¶
metric的计算
- Parameters:
metric – str, 自带metrics的名称
func – function, 透传的用户自定的计算指标的函数
y_pred – torch.Tensor, 样本的预测结果
y_true – torch.Tensor, 样本的真实结果
- torch4keras.snippets.search_layer(model, layer_name, retrun_first=True)[source]¶
根据layer_name搜索并返回参数/参数list
- torch4keras.snippets.send_email(receivers, subject, msg='', mail_host=None, mail_user=None, mail_pwd=None, sender=None)[source]¶
发送邮件(默认使用笔者自己注册的邮箱，若含敏感信息请使用自己注册的邮箱)
- Parameters:
subject – str, 邮件主题
msg – str, 邮件正文
receivers – str/list, 收件人邮箱
mail_host – str, 发件服务器host
mail_user – str, 发件人
mail_pwd – str, smtp的第三方密码
sender – str, 发件人邮箱