search

guillaume-chevalier / LSTM-Human-Activity-Recognition

Human Activity Recognition example using TensorFlow on smartphone sensors dataset and an LSTM RNN. Classifying the type of movement amongst six activity categories - Guillaume Chevalier
MIT License
3.33k stars 935 forks source link

Clean pipeline using Neuraxle #33

Open guillaume-chevalier opened 4 years ago

guillaume-chevalier commented 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.