Systems, apparatus and methods are provided for determining whether data associated with a command in a storage system is hot or cold. An apparatus may comprise a first interface to be coupled to a host and circuitry configured to receive a command via the first interface from the host. The command may contain an address for data storage in a data storage system and a length for a size of data associated with the command. The circuitry may be further configured to classify the command into a cluster of a plurality of clusters, obtain a plurality of historical commands in the cluster and determine whether the data associated with the command is hot or cold using a machine learning model. The address and the length of the command and addresses and lengths of the plurality of historical commands may be used as input features to the machine learning model.