Preliminary open source and interpretation of the blog: blog.csdn.net/xiaosongshi…

Heats, blog blog.csdn.net/xiaosongshi…

The code consists of two parts: 1. Training code and 2. Prediction code

1. Training code, KFold is used, weighted loss and residual connection

from glob import glob
import pandas as pd
import numpy as np
from scipy.io import loadmat
import math
from matplotlib import pyplot as plt
from keras.layers import *
from keras.models import *
from keras.optimizers import *
import keras
from keras import backend as K
#import tqdm
import random
import tensorflow as tf

seed = 1234
random.seed(seed)
tf.set_random_seed(seed)



base_path = "/media/jdcloud/"
train_base_dir = base_path+"Train/"
val_base_dir = base_path+"Val/"

ref_path = base_p
Copy the code