Learning to infer weather states using partial observations