Skip to content

Clean pipeline using Neuraxle #33

Description

@guillaume-chevalier

Should do something that looks like this to clean the project by using Neuraxle:

deep_learning_seq_classif_pipeline = EpochRepeater(Pipeline([
    TrainOnlyWrapper(DataShuffler(seed=42)),
    MiniBatchSequentialPipeline([
        ForEachDataInput(Pipeline([
            ToNumpy(np_dtype=np.float32),
            DefaultValuesFiller(0.0),
        ])),
        ClassificationLSTM(n_stacked=2, n_residual=3),
    ], batch_size=32),
]), epochs=200, fit_only=True)

Where the ClassificationLSTM class contains the actual TensorFlow Code.

Metadata

Metadata

Assignees

No one assigned

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions