python - PySpark reversing StringIndexer in nested array -


i'm using pyspark collaborative filtering using als. original user , item id's strings, used stringindexer convert them numeric indices (pyspark's als model obliges so).

after i've fitted model, can top 3 recommendations each user so:

recs = (     model     .recommendforallusers(3) )

the recs dataframe looks so:

+-----------+--------------------+ |useridindex|     recommendations| +-----------+--------------------+ |       1580|[[10096,3.6725707...| |       4900|[[10096,3.0137873...| |       5300|[[10096,2.7274625...| |       6620|[[10096,2.4493625...| |       7240|[[10096,2.4928937...| +-----------+--------------------+ showing top 5 rows  root  |-- useridindex: integer (nullable = false)  |-- recommendations: array (nullable = true)  |    |-- element: struct (containsnull = true)  |    |    |-- productidindex: integer (nullable = true)  |    |    |-- rating: float (nullable = true)

i want create huge jsom dump dataframe, , can so:

(     recs     .tojson()     .saveastextfile("name_i_must_hide.recs") )

and sample of these jsons is:

{   "useridindex": 1580,   "recommendations": [     {       "productidindex": 10096,       "rating": 3.6725707     },     {       "productidindex": 10141,       "rating": 3.61542     },     {       "productidindex": 11591,       "rating": 3.536216     }   ] }

the useridindex , productidindex keys due stringindexer transformation.

how can original value of these columns back? suspect must use indextostring transformer, can't quite figure out how since data nested in array inside recs dataframe.

i tried use pipeline evaluator (stages=[stringindexer, als, indextostring]) looks evaluator doesn't support these indexers.

cheers!

in both cases you'll need access list of labels. can accessed using either stringindexermodel

user_indexer_model = ...  # type: stringindexermodel user_labels = user_indexer_model.labels  product_indexer_model = ...  # type: stringindexermodel product_labels = product_indexer_model.labels

or column metadata.

for useridindex can apply indextostring:

from pyspark.ml.feature import indextostring  user_id_to_label = indextostring(     inputcol="useridindex", outputcol="userid", labels=user_labels) user_id_to_label.transform(recs)

for recommendations you'll need either udf or expression this:

from pyspark.sql.functions import array, col, lit, struct  n = 3  # same numitems  product_labels_ = array(*[lit(x) x in product_labels]) recommendations = array(*[struct(     product_labels_[col("recommendations")[i]["productidindex"]].alias("productid"),     col("recommendations")[i]["rating"].alias("rating") ) in range(n)])  recs.withcolumn("recommendations", recommendations)

Comments

Post a Comment

Popular posts from this blog

python - Operations inside variables -

this question has answer here: evaluating mathematical expression in string 12 answers i want make python script finds maximum possible output using 2 operations. problem says there syntax error @ 2nd last line, in b in there. how can fix it? x = 0 = int(input("1st number:")) c = int(input("2nd number:" )) e = int(input("3rd number:" )) in range(4): if == 0: b = "+" elif == 1: b = "-" elif == 2: b = "/" else: b = "*" j in range(4): if j == 0: d = "+" elif j == 1: d = "-" elif j == 2: d = "/" else: d = "*" k = b c d e print(k) if you're after variables e on single line, couldn...
Read more

Generic Map Parameter java -

i having lot of methods of type: public static list<eduusers> getdetails(class c, map<string, integer> params) { list<eduusers> details = null; session session = hibernateutil.getsessionfactory().opensession(); transaction tx = null; try { tx = session.begintransaction(); criteria cr = session.createcriteria(c); (map.entry<string, integer> entry : params.entryset()) { cr.add(restrictions.eq(entry.getkey(), entry.getvalue())); } details = cr.list(); tx.commit(); } catch (exception asd) { log.debug(asd.getmessage()); if (tx != null) { tx.commit(); } } { session.close(); } return details; } i trying have generic method them , have written far: public static <t> list<t> getdetails(class<t> c, class<t> b) { l...
Read more

arrays - What causes a java.lang.ArrayIndexOutOfBoundsException and how do I prevent it? -

what arrayindexoutofboundsexception mean , how rid of it? here code sample triggers exception: string[] name = {"tom", "dick", "harry"}; for(int = 0; i<=name.length; i++) { system.out.print(name[i] +'\n'); } your first port of call should documentation explains reasonably clearly: thrown indicate array has been accessed illegal index. index either negative or greater or equal size of array. so example: int[] array = new int[5]; int boom = array[10]; // throws exception as how avoid it... um, don't that. careful array indexes. one problem people run thinking arrays 1-indexed, e.g. int[] array = new int[5]; // ... populate array here ... (int index = 1; index <= array.length; index++) { system.out.println(array[index]); } that miss out first element (index 0) , throw exception when index 5. valid indexes here 0-4 inclusive. correct, idiomatic for statement here be: for (int index = 0; index <...
Read more