pythonの関数定義位置（美しい書き方）

Question

自分のpythonコードが醜くて仕方ありません。
コーディング規約はQiita等でまとめて下さってたページを見ましたが、参考になる情報は見つけられませんでした。

私の場合、関数と実行文があちこちに記載され実行文を見落としてしまいます（視認性が悪い）。
かと言って関数を一箇所にまとめるとpythonのスクリプト感がなくなるしなぁと悩んでます。
皆様の経験か、コーディング規約等があればご紹介ください。

DATA_SET = 'path/to_image'
num_epochs = 100

def make_dataset_path(root):
    img_path = os.path.join(root, 'images')
    label_path = os.path.join(root, 'labelss')
    dataset = hogehoge
    return dataset

dataset = make_dataset(DATA_SET)
dataloader = torch.utils.data.DataLoader(dataset, batch_size=batch_size,
                                         shuffle=True, num_workers=2)

def weights_init(m):
    hogehoge

model = Net()　　　　　　　　　　　　　←　このように関数の間に実行文があるのが気持ち悪い
model.apply(weight_init)　　　　　　 ←　関数たちを上の方にまとめて記述するべき？　

def train(epoch):
    for image, label in dataloader:
        hogehoge

for epoch in range(num_epochs):
    train(epoch) 

Answer 1

関数定義は上の方にまとめ、ついでに一番下にmain関数を書いてそのスクリプトで行う処理を押し込み、その下でif __name__ == "__main__":とか使って呼び出すのが割と綺麗でよく見るスタイルのように思います。