focal.commands.train_commands module¶

Module for defining training logic to call from main entry point

class focal.commands.train_commands.KFoldCNN[source]¶

Train CNN model using k-fold cross validation.

class focal.commands.train_commands.KFoldMLP[source]¶

Train MLP model using k-fold cross validation.

class focal.commands.train_commands.TrainCNN[source]¶

Train a CNN model for fiber cleave classification.

class focal.commands.train_commands.TrainCustomModel[source]¶

Train a custom CNN model without pre-trained base.

class focal.commands.train_commands.TrainImageOnly[source]¶

Train the CNN model with only images.

class focal.commands.train_commands.TrainMLP[source]¶

Train an MLP model for tension prediction.

class focal.commands.train_commands.TrainRL[source]¶

Train the reinforement learning agent

class focal.commands.train_commands.TrainXGBoost[source]¶

Train an XGBoost model for predicting delta in tension