init model

Init solo model object with given hyperparameters.

Parameters

device: str Device to run the training on (CPU or Cuda if GPU is available).

epochs: str Number of epochs to train models for.

learning_rate: float Adjustable learning rate for improving gradient descent opttimizer.

train_size: int The percentage of the dataset to be used for training (not testing/validation).

Web view

init_model_screenshot