import tensorflow as tf

class LSTM_Classifier(tf.keras.Model):
    def __init__(self, lstm_units=128, num_classes=4, **kwargs):
        super(LSTM_Classifier, self).__init__(**kwargs)
        self.lstm = tf.keras.layers.LSTM(lstm_units, return_sequences=False)
        self.dropout = tf.keras.layers.Dropout(0.3)
        self.classifier = tf.keras.layers.Dense(num_classes, activation='softmax')

    def call(self, inputs):
        x = self.lstm(inputs)
        x = self.dropout(x)
        return self.classifier(x)