xlearn.classify
¶
Module containing model, predict and train routines
Functions:
model (dim_img, nb_filters, nb_conv, nb_classes) |
the cnn model for image transformation |
train (x_train, y_train, x_test, y_test, …) |
Function description. |
-
xlearn.classify.
model
(dim_img, nb_filters, nb_conv, nb_classes)[source]¶ the cnn model for image transformation
Parameters: - dim_img (int) – The input image dimension
- nb_filters (int) – Number of filters
- nb_conv (int) – The convolution weight dimension
Returns: mdl – Description.
-
xlearn.classify.
train
(x_train, y_train, x_test, y_test, dim_img, nb_filters, nb_conv, batch_size, nb_epoch, nb_classes)[source]¶ Function description.
Parameters: - parameter_01 (type) – Description.
- parameter_02 (type) – Description.
- parameter_03 (type) – Description.
Returns: return_01 – Description.