tsai

tsai copied to clipboard

I am using the TSClassifier to implement the TST model for time series classification. Every time I run the code, I have been receiving the error below:

/usr/local/lib/python3.10/dist-packages/tsai/models/TST.py in init(self, c_in, c_out, seq_len, max_seq_len, n_layers, d_model, n_heads, d_k, d_v, d_ff, dropout, act, fc_dropout, y_range, verbose, **kwargs) 172 self.new_q_len = True 173 t = torch.rand(1, 1, seq_len) --> 174 q_len = nn.Conv1d(1, 1, **kwargs)(t).shape[-1] 175 self.W_P = nn.Conv1d(c_in, d_model, **kwargs) # Eq 2 176 pv(f'Conv1d with kwargs={kwargs} applied to input to create input encodings\n', verbose)

TypeError: Conv1d.init() got an unexpected keyword argument 'custom_head'