KMJcustom/utils.py
def shape(a, w, f):
    '''
    a: number of stocks
    w: window size
    f: number of features
    '''
    SHAPE={}
    SHAPE['flatten']=(a*w*f, )
    SHAPE['awf']=(a, w, f)
    SHAPE['waf']=(w, a, f)
    SHAPE['wfa']=(w, f, a)
    SHAPE['fwa']=(f, w, a)
    return SHAPE

    a: number of stocks
    w: window size
    f: number of features
    
number of stocks
w: window size
f: number of features
number of stocks w : window size f : number of features
def shape(a, w, f):
    SHAPE={}
    SHAPE['flatten']=(a*w*f, )
    SHAPE['awf']=(a, w, f)
    SHAPE['waf']=(w, a, f)
    SHAPE['wfa']=(w, f, a)
    SHAPE['fwa']=(f, w, a)
    return SHAPE
def shape ( a , w , f ) : SHAPE = { } SHAPE [ 'flatten' ] = ( a * w * f , ) SHAPE [ 'awf' ] = ( a , w , f ) SHAPE [ 'waf' ] = ( w , a , f ) SHAPE [ 'wfa' ] = ( w , f , a ) SHAPE [ 'fwa' ] = ( f , w , a ) return SHAPE
a: number of stocks
w: window size
f: number of features
a : number of stocks w : window size f : number of features
'''
    a: number of stocks
    w: window size
    f: number of features
    '''
a
w
f