caf.brain.ml

caf.brAIn is Transport for the North’s bespoke machine learning and AI library. It consists of a generalised end to end machine learning pipeline, simplified machine learning functions and a machine vision model.

Functions

algorithm_evaluation(model_choice, data, ...)	Evaluate which algorithm is best performing.
evaluate_data(data, output_path, ...[, ...])	Data analysis for structured tabular numerical data.
feature_selection(data, output_path, ...[, ...])	Conduct simple feature selection.
hparam_optim(model_choice, data, ...[, ...])	Hyperparameter optimisation for your selected algorithm.
tidy_data(data_path, ...[, custom_index, ...])	Converts semi-structured data into structured for machine learning.
transform_data(data, output_path, ...[, ...])	Encode and or scale data where applicable for machine learning modelling.

Classes

Models(value)

Available algorithms for the caf.brAIn prediction model.