From 5c1bad3589c247f8353b86fbb3bc28afafab5a8d Mon Sep 17 00:00:00 2001 From: guanjihuan Date: Tue, 28 Nov 2023 21:28:44 +0800 Subject: [PATCH] 0.1.55 --- PyPI/setup.cfg | 2 +- PyPI/src/guan.egg-info/PKG-INFO | 2 +- PyPI/src/guan/machine_learning.py | 4 ++++ 3 files changed, 6 insertions(+), 2 deletions(-) diff --git a/PyPI/setup.cfg b/PyPI/setup.cfg index 5fd5b8c..6c5fbfe 100644 --- a/PyPI/setup.cfg +++ b/PyPI/setup.cfg @@ -1,7 +1,7 @@ [metadata] # replace with your username: name = guan -version = 0.1.54 +version = 0.1.55 author = guanjihuan author_email = guanjihuan@163.com description = An open source python package diff --git a/PyPI/src/guan.egg-info/PKG-INFO b/PyPI/src/guan.egg-info/PKG-INFO index dae4294..ffdbdf1 100644 --- a/PyPI/src/guan.egg-info/PKG-INFO +++ b/PyPI/src/guan.egg-info/PKG-INFO @@ -1,6 +1,6 @@ Metadata-Version: 2.1 Name: guan -Version: 0.1.54 +Version: 0.1.55 Summary: An open source python package Home-page: https://py.guanjihuan.com Author: guanjihuan diff --git a/PyPI/src/guan/machine_learning.py b/PyPI/src/guan/machine_learning.py index 3091100..d8e4d1b 100644 --- a/PyPI/src/guan/machine_learning.py +++ b/PyPI/src/guan/machine_learning.py @@ -1,10 +1,12 @@ # Module: machine_learning import guan +model_class = None # 把类定义成全局的，防止保存完整模型时，无法访问函数中的类 # 全连接神经网络模型（包含一个隐藏层） @guan.function_decorator def fully_connected_neural_network_with_one_hidden_layer(input_size=1, hidden_size=10, output_size=1, activation='relu'): import torch + global model_class class model_class(torch.nn.Module): def __init__(self): super().__init__() @@ -30,6 +32,7 @@ def fully_connected_neural_network_with_one_hidden_layer(input_size=1, hidden_si @guan.function_decorator def fully_connected_neural_network_with_two_hidden_layers(input_size=1, hidden_size_1=10, hidden_size_2=10, output_size=1, activation_1='relu', activation_2='relu'): import torch + global model_class class model_class(torch.nn.Module): def __init__(self): super().__init__() @@ -68,6 +71,7 @@ def fully_connected_neural_network_with_two_hidden_layers(input_size=1, hidden_s @guan.function_decorator def fully_connected_neural_network_with_three_hidden_layers(input_size=1, hidden_size_1=10, hidden_size_2=10, hidden_size_3=10, output_size=1, activation_1='relu', activation_2='relu', activation_3='relu'): import torch + global model_class class model_class(torch.nn.Module): def __init__(self): super().__init__()