Open guillaume-chevalier opened 4 years ago
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.
ClassificationLSTM
Should do something that looks like this to clean the project by using Neuraxle:
Where the
ClassificationLSTM
class contains the actual TensorFlow Code.