search

amazon-science / earth-forecasting-transformer

Official implementation of Earthformer
Apache License 2.0
349 stars 58 forks source link

Questions about input and output #71

Open fizzking opened 2 months ago

fizzking commented 2 months ago

Can earthformer be used for multivariate prediction of univariate? For example, my input variable is [T,H,W,5], and my output variable is some other quantity [T,H,W,1]？

gaozhihan commented 2 months ago

Sure. You may want to have a try in our simplest test case to set different values for the input channels and target channels via input_shape and target_shape.