model.py

import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd import Function

import torch.nn as nn

class LinearRegression(nn.Module):
    def __init__(self, cfg, input_size):
        super(LinearRegression, self).__init__()
        self.input_size = input_size
        self.bias = cfg["LinearRegression"]["Bias"]
        self.regressor = nn.Linear(input_size, 1, self.bias)
        self.activate = nn.ReLU()
    def forward(self, x):
        y = self.regressor(x).view((x.shape[0],-1))
        y = self.activate(y)
        return y

def build_model(cfg, input_size):    
    return eval(f"{cfg['Model']['Name']}(cfg, input_size)")


if __name__== "__main__":
    import yaml
    config_file = open("config.yml","r")
    cfg = yaml.load(config_file)
    print(cfg['Model']['Name'])