focal.commands.test_commands module

This module define the logic for testing each model.

class focal.commands.test_commands.TestCNN[source]

Bases: BaseCommand

Test CNN model performance.

class focal.commands.test_commands.TestImageOnly[source]

Bases: BaseCommand

Test CNN model performance on only images.

class focal.commands.test_commands.TestMLP[source]

Bases: BaseCommand

Test MLP model performance.

class focal.commands.test_commands.TestRL[source]

Bases: BaseCommand

Testing logic for RL agent

class focal.commands.test_commands.TestXGBoost[source]

Bases: BaseCommand

Test XGBoost performance on prediciting change in tension.